taskcluster-taskgraph 11.2.0__tar.gz → 11.2.1__tar.gz

{taskcluster_taskgraph-11.2.0 → taskcluster_taskgraph-11.2.1}/CHANGELOG.md

@@ -1,5 +1,11 @@
 # Change Log
 
+## [11.2.1] - 2024-10-03
+
+### Fixed
+
+- `taskgraph.optimize.register_strategy` can now pass kwargs on to the optimization class
+
 ## [11.2.0] - 2024-09-26
 
 ### Added

{taskcluster_taskgraph-11.2.0 → taskcluster_taskgraph-11.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: taskcluster-taskgraph
-Version: 11.2.0
+Version: 11.2.1
 Summary: Build taskcluster taskgraphs
 Project-URL: Repository, https://github.com/taskcluster/taskgraph
 Project-URL: Issues, https://github.com/taskcluster/taskgraph/issues

{taskcluster_taskgraph-11.2.0 → taskcluster_taskgraph-11.2.1}/src/taskgraph/__init__.py

@@ -2,7 +2,7 @@
 # License, v. 2.0. If a copy of the MPL was not distributed with this
 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
-__version__ = "11.2.0"
+__version__ = "11.2.1"
 
 # Maximum number of dependencies a single task can have
 # https://docs.taskcluster.net/docs/reference/platform/queue/api#createTask

{taskcluster_taskgraph-11.2.0 → taskcluster_taskgraph-11.2.1}/src/taskgraph/optimize/base.py

@@ -28,10 +28,12 @@ logger = logging.getLogger("optimization")
 registry = {}
 
 
-def register_strategy(name, args=()):
+def register_strategy(name, args=(), kwargs=None):
+    kwargs = kwargs or {}
+
     def wrap(cls):
         if name not in registry:
-            registry[name] = cls(*args)
+            registry[name] = cls(*args, **kwargs)
             if not hasattr(registry[name], "description"):
                 registry[name].description = name
         return cls

{taskcluster_taskgraph-11.2.0 → taskcluster_taskgraph-11.2.1}/test/test_optimize.py

@@ -9,7 +9,13 @@ import pytest
 
 from taskgraph.graph import Graph
 from taskgraph.optimize import base as optimize_mod
-from taskgraph.optimize.base import All, Any, Not, OptimizationStrategy
+from taskgraph.optimize.base import (
+    All,
+    Any,
+    Not,
+    OptimizationStrategy,
+    register_strategy,
+)
 from taskgraph.task import Task
 from taskgraph.taskgraph import TaskGraph
 
@@ -467,3 +473,10 @@ def test_get_subgraph_removed_dep():
     graph = make_triangle()
     with pytest.raises(Exception):
         optimize_mod.get_subgraph(graph, {"t2"}, set(), {})
+
+
+def test_register_strategy(mocker):
+    m = mocker.Mock()
+    func = register_strategy("foo", args=("one", "two"), kwargs={"n": 1})
+    func(m)
+    m.assert_called_with("one", "two", n=1)