Browse Source

feat: [TTA] _sort_by_score_time()

Pull out a one-liner for test scenarios to facilitate testing adjust()
main
Rob Hallam 1 month ago
parent
commit
d0935f950f
1 changed files with 10 additions and 0 deletions
  1. +10
    -0
      pipeline/adjusters.py

+ 10
- 0
pipeline/adjusters.py View File

@@ -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…
Cancel
Save