hydraflow 0.4.2__tar.gz → 0.4.3__tar.gz

{hydraflow-0.4.2 → hydraflow-0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: hydraflow
-Version: 0.4.2
+Version: 0.4.3
 Summary: Hydraflow integrates Hydra and MLflow to manage and track machine learning experiments.
 Project-URL: Documentation, https://github.com/daizutabi/hydraflow
 Project-URL: Source, https://github.com/daizutabi/hydraflow

{hydraflow-0.4.2 → hydraflow-0.4.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "hydraflow"
-version = "0.4.2"
+version = "0.4.3"
 description = "Hydraflow integrates Hydra and MLflow to manage and track machine learning experiments."
 readme = "README.md"
 license = "MIT"

{hydraflow-0.4.2 → hydraflow-0.4.3}/src/hydraflow/run_collection.py

@@ -98,6 +98,32 @@ class RunCollection:
     def __bool__(self) -> bool:
         return bool(self._runs)
 
+    def __add__(self, other: RunCollection) -> RunCollection:
+        """Add another `RunCollection` to this one.
+
+        Args:
+            other (RunCollection): The `RunCollection` to add.
+
+        Returns:
+            A new `RunCollection` instance with the runs from both collections.
+
+        """
+        return self.__class__(self._runs + other._runs)
+
+    def __sub__(self, other: RunCollection) -> RunCollection:
+        """Subtract another `RunCollection` from this one.
+
+        Args:
+            other (RunCollection): The `RunCollection` to subtract.
+
+        Returns:
+            A new `RunCollection` instance with the runs that are in this collection
+            but not in the other.
+
+        """
+        runs = [run for run in self._runs if run not in other._runs]  # noqa: SLF001
+        return self.__class__(runs)
+
     @classmethod
     def from_list(cls, runs: list[Run]) -> RunCollection:
         """Create a `RunCollection` instance from a list of MLflow `Run` instances."""

{hydraflow-0.4.2 → hydraflow-0.4.3}/tests/test_run_collection.py

@@ -49,6 +49,20 @@ def test_from_list(run_list: list[Run]):
     assert all(run in rc for run in run_list)
 
 
+def test_add(run_list: list[Run]):
+    rc1 = RunCollection.from_list(run_list[:3])
+    rc2 = RunCollection.from_list(run_list[3:])
+    rc = rc1 + rc2
+    assert rc._runs == run_list
+
+
+def test_sub(run_list: list[Run]):
+    rc1 = RunCollection.from_list(run_list)
+    rc2 = RunCollection.from_list(run_list[3:])
+    rc = rc1 - rc2
+    assert rc._runs == run_list[:3]
+
+
 def test_search_runs_sorted(run_list: list[Run]):
     assert [run.data.params["p"] for run in run_list] == ["0", "1", "2", "3", "4", "5"]