Bug 1635149 - Force scheduling tasks which were not known when the bugbug test selection model was trained. r=ahal

Differential Revision: https://phabricator.services.mozilla.com/D73718

taskcluster/taskgraph/optimize/bugbug.py

@@ -54,6 +54,9 @@ class BugBugPushSchedules(OptimizationStrategy):
 
         test_manifests = task.attributes.get('test_manifests')
         if test_manifests is None or self.use_reduced_tasks:
+            if "known_tasks" in data and task.label not in data["known_tasks"]:
+                return False
+
             if task.label not in tasks:
                 return True
 

taskcluster/taskgraph/test/test_optimize_strategies.py

@@ -160,6 +160,13 @@ def test_optimization_strategy(responses, params, opt, tasks, arg, expected):
         ['task-2'],
     ),
 
+    # tasks which are unknown to bugbug are selected
+    pytest.param(
+        (0.1,),
+        {'tasks': {'task-1': 0.9, 'task-3': 0.5}, 'known_tasks': ['task-1', 'task-3', 'task-4']},
+        ['task-2'],
+    ),
+
     # tasks containing groups selected
     pytest.param(
         (0.1,),