瀏覽代碼

feat: [TTA] _sort_by_score_time()

Pull out a one-liner for test scenarios to facilitate testing adjust()
main
Rob Hallam 2 月之前
父節點
當前提交
d0935f950f
共有 1 個文件被更改,包括 10 次插入0 次删除
  1. +10
    -0
      pipeline/adjusters.py

+ 10
- 0
pipeline/adjusters.py 查看文件

@@ -70,6 +70,16 @@ class TargetTimeAdjuster(Adjuster):
"""
return float(sum([x.interval.duration for x in features]))

def _sort_by_score_time(self, features: list) -> list:
"""Sort Features by score (primary) and by time (secondary).

Returns a sorted list of Features.

Pulled out for unit testing as RDH was having issues with adjust()
and wanted to verify sorting was working correctly.
"""
return sorted(features, key=lambda x: (x.score, x.interval.duration))

def __init__(self, features: list=[],
target_time: int|float=_DEFAULT_TARGET_TIME,
margin: int|float=_DEFAULT_MARGIN,


Loading…
取消
儲存