hydraflow 0.4.2__tar.gz → 0.4.4__tar.gz

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

@@ -1,12 +1,12 @@
 Metadata-Version: 2.3
 Name: hydraflow
-Version: 0.4.2
+Version: 0.4.4
 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
 Project-URL: Issues, https://github.com/daizutabi/hydraflow/issues
 Author-email: daizutabi <daizutabi@gmail.com>
-License-Expression: MIT
+License: MIT
 License-File: LICENSE
 Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python

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

@@ -4,10 +4,10 @@ build-backend = "hatchling.build"
 
 [project]
 name = "hydraflow"
-version = "0.4.2"
+version = "0.4.4"
 description = "Hydraflow integrates Hydra and MLflow to manage and track machine learning experiments."
 readme = "README.md"
-license = "MIT"
+license = { text = "MIT" }
 authors = [{ name = "daizutabi", email = "daizutabi@gmail.com" }]
 classifiers = [
   "Development Status :: 4 - Beta",

{hydraflow-0.4.2 → hydraflow-0.4.4}/src/hydraflow/context.py

@@ -28,7 +28,7 @@ log = logging.getLogger(__name__)
 
 @contextmanager
 def log_run(
-    config: object,
+    config: object | None,
     *,
     synchronous: bool | None = None,
 ) -> Iterator[None]:
@@ -54,7 +54,8 @@ def log_run(
         ```
 
     """
-    log_params(config, synchronous=synchronous)
+    if config:
+        log_params(config, synchronous=synchronous)
 
     hc = HydraConfig.get()
     output_dir = Path(hc.runtime.output_dir)
@@ -139,7 +140,7 @@ def start_run(  # noqa: PLR0913
             description=description,
             log_system_metrics=log_system_metrics,
         ) as run,
-        log_run(config, synchronous=synchronous),
+        log_run(config if run_id is None else None, synchronous=synchronous),
     ):
         yield run
 

{hydraflow-0.4.2 → hydraflow-0.4.4}/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.4}/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"]