feat: prioritize proven GPU framework combinations

tests/test_gpu_strategy.py

@@ -14,7 +14,6 @@ if str(PACKAGE_DIR) in sys.path:
 sys.path.insert(0, str(PACKAGE_DIR))
 
 from gpu_strategy import (  # noqa: E402
-    ALL_SUPPORTED,
     LONG_TERM,
     RECENT,
     GPUStrategyManager,
@@ -55,7 +54,15 @@ class WeightedMarket:
         return {"gpu-a": 3.0, "gpu-b": 0.1, "gpu-c": 0.1, "gpu-d": 0.1}[gpu]
 
     def gpu_metadata(self, gpu: str) -> dict:
-        return {"marketWeight": self.gpu_weight(gpu, category=ALL_SUPPORTED)}
+        return {"marketWeight": self.gpu_weight(gpu, category=LONG_TERM)}
+
+    def eligible_gpus(self, supported_gpus: list[str]) -> list[str]:
+        return list(supported_gpus)
+
+
+class VettedMarket(WeightedMarket):
+    def eligible_gpus(self, supported_gpus: list[str]) -> list[str]:
+        return [gpu for gpu in supported_gpus if gpu in {"gpu-b", "gpu-c"}]
 
 
 class GPUStrategyTests(unittest.TestCase):
@@ -80,14 +87,14 @@ class GPUStrategyTests(unittest.TestCase):
         self.assertNotIn("tiny", snapshot["longTermGpus"])
 
     def test_weighted_category_planner_is_exact_over_200_accepts(self) -> None:
-        counts = {LONG_TERM: 0, ALL_SUPPORTED: 0, RECENT: 0}
+        counts = {LONG_TERM: 0, RECENT: 0}
         for _ in range(200):
             category = choose_next_category(counts)
             counts[category] += 1
 
-        self.assertEqual({LONG_TERM: 100, ALL_SUPPORTED: 60, RECENT: 40}, counts)
+        self.assertEqual({LONG_TERM: 140, RECENT: 60}, counts)
 
-    def test_candidate_order_uses_50_30_20_for_first_strategy_generation(self) -> None:
+    def test_candidate_order_uses_70_30_without_exploration(self) -> None:
         supported = ["gpu-a", "gpu-b", "gpu-c", "gpu-d"]
         snapshot = build_strategy_snapshot([], supported_gpus=supported)
         manager = GPUStrategyManager("unused.json")
@@ -101,9 +108,9 @@ class GPUStrategyTests(unittest.TestCase):
         ordered = manager.order_candidates(candidates)
         categories = Counter(candidate["strategyCategory"] for candidate in ordered[:200])
 
-        self.assertEqual(100, categories[LONG_TERM])
-        self.assertEqual(60, categories[ALL_SUPPORTED])
-        self.assertEqual(40, categories[RECENT])
+        self.assertEqual(140, categories[LONG_TERM])
+        self.assertEqual(60, categories[RECENT])
+        self.assertNotIn("all_supported", categories)
 
     def test_refresh_happens_only_after_200_accepted_submissions(self) -> None:
         with tempfile.TemporaryDirectory() as temporary_dir:
@@ -126,7 +133,7 @@ class GPUStrategyTests(unittest.TestCase):
             self.assertEqual(200, manager.state["acceptedTotal"])
             self.assertEqual(0, manager.state["acceptedSinceRefresh"])
 
-    def test_market_weights_change_gpu_mix_without_changing_50_30_20_split(self) -> None:
+    def test_market_weights_change_gpu_mix_without_changing_70_30_split(self) -> None:
         supported = ["gpu-a", "gpu-b", "gpu-c", "gpu-d"]
         manager = GPUStrategyManager("unused.json", market_intelligence=WeightedMarket())
         manager.state = build_strategy_snapshot([], supported_gpus=supported)
@@ -138,15 +145,26 @@ class GPUStrategyTests(unittest.TestCase):
 
         ordered = manager.order_candidates(candidates)[:200]
         category_counts = Counter(candidate["strategyCategory"] for candidate in ordered)
-        exploration_counts = Counter(
-            candidate["targetGpu"]
-            for candidate in ordered
-            if candidate["strategyCategory"] == ALL_SUPPORTED
-        )
+        gpu_counts = Counter(candidate["targetGpu"] for candidate in ordered)
 
-        self.assertEqual({LONG_TERM: 100, ALL_SUPPORTED: 60, RECENT: 40}, category_counts)
-        self.assertGreater(exploration_counts["gpu-a"], exploration_counts["gpu-b"])
-        self.assertTrue(all(exploration_counts[gpu] > 0 for gpu in supported))
+        self.assertEqual({LONG_TERM: 140, RECENT: 60}, category_counts)
+        self.assertGreater(gpu_counts["gpu-a"], gpu_counts["gpu-b"])
+        self.assertNotIn("all_supported", category_counts)
+
+    def test_unvetted_gpus_receive_zero_submissions(self) -> None:
+        supported = ["gpu-a", "gpu-b", "gpu-c", "gpu-d"]
+        manager = GPUStrategyManager("unused.json", market_intelligence=VettedMarket())
+        manager.state = build_strategy_snapshot([], supported_gpus=supported)
+        candidates = [
+            {"repoId": f"owner/model-{model}", "targetGpu": gpu}
+            for model in range(50)
+            for gpu in supported
+        ]
+
+        ordered = manager.order_candidates(candidates)
+
+        self.assertTrue(ordered)
+        self.assertEqual({"gpu-b", "gpu-c"}, {candidate["targetGpu"] for candidate in ordered})
 
 
 if __name__ == "__main__":