Adjust OrderHistory.choose_tactical_model() heuristic
- use the `HorizontalSMAModel` for low demand - use the `TrivialModel` for no demand
This commit is contained in:
parent
3f5b4a50bb
commit
23391c2fa4
2 changed files with 4 additions and 8 deletions
|
|
@ -112,8 +112,7 @@ class TestChooseTacticalModel:
|
|||
train_horizon=test_config.LONG_TRAIN_HORIZON,
|
||||
)
|
||||
|
||||
# TODO: this should be the future `HorizontalSMAModel`.
|
||||
assert isinstance(result, models.HorizontalETSModel)
|
||||
assert isinstance(result, models.HorizontalSMAModel)
|
||||
|
||||
def test_best_model_with_no_demand(
|
||||
self, order_history, good_pixel_id, predict_at,
|
||||
|
|
@ -127,8 +126,7 @@ class TestChooseTacticalModel:
|
|||
train_horizon=test_config.LONG_TRAIN_HORIZON,
|
||||
)
|
||||
|
||||
# TODO: this should be the future `HorizontalTrivialModel`.
|
||||
assert isinstance(result, models.HorizontalETSModel)
|
||||
assert isinstance(result, models.TrivialModel)
|
||||
|
||||
def test_best_model_for_unknown_train_horizon(
|
||||
self, order_history, good_pixel_id, predict_at, # noqa:RST215
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue