feat: make model age admission-only
This commit is contained in:
@@ -1,6 +1,5 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import json
|
||||
import sys
|
||||
import tempfile
|
||||
@@ -18,7 +17,7 @@ from outcome_tracker import OutcomeTracker # noqa: E402
|
||||
from poll_runner import ( # noqa: E402
|
||||
_bootstrap_architecture_history,
|
||||
_load_task_compatibility_contexts,
|
||||
resolve_age_cleanup_policy,
|
||||
build_parser,
|
||||
)
|
||||
|
||||
|
||||
@@ -107,31 +106,11 @@ class PollPolicyTests(unittest.TestCase):
|
||||
self.assertEqual("mindie", contexts["task-old"]["framework"])
|
||||
self.assertEqual({}, contexts["task-old"]["modelProfile"])
|
||||
|
||||
def test_age_cleanup_uses_minus_ten_once_then_minus_five(self) -> None:
|
||||
args = argparse.Namespace(
|
||||
recent_model_reserve_slots=10,
|
||||
dynamic_old_model_cleanup_reserve_slots=5,
|
||||
)
|
||||
def test_age_policy_defaults_to_admission_only_reserve_five(self) -> None:
|
||||
args = build_parser().parse_args([])
|
||||
|
||||
self.assertEqual(
|
||||
("initial", 10),
|
||||
resolve_age_cleanup_policy(args, initial_cleanup_pending=True),
|
||||
)
|
||||
self.assertEqual(
|
||||
("dynamic", 5),
|
||||
resolve_age_cleanup_policy(args, initial_cleanup_pending=False),
|
||||
)
|
||||
|
||||
def test_dynamic_cleanup_cannot_be_stricter_than_initial_cleanup(self) -> None:
|
||||
args = argparse.Namespace(
|
||||
recent_model_reserve_slots=10,
|
||||
dynamic_old_model_cleanup_reserve_slots=20,
|
||||
)
|
||||
|
||||
self.assertEqual(
|
||||
("dynamic", 10),
|
||||
resolve_age_cleanup_policy(args, initial_cleanup_pending=False),
|
||||
)
|
||||
self.assertEqual(5, args.recent_model_reserve_slots)
|
||||
self.assertFalse(hasattr(args, "dynamic_old_model_cleanup_reserve_slots"))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
Reference in New Issue
Block a user