hydraflow 0.5.2__tar.gz → 0.5.3__tar.gz

{hydraflow-0.5.2 → hydraflow-0.5.3}/PKG-INFO

@@ -1,8 +1,8 @@
 Metadata-Version: 2.4
 Name: hydraflow
-Version: 0.5.2
+Version: 0.5.3
 Summary: Hydraflow integrates Hydra and MLflow to manage and track machine learning experiments.
-Project-URL: Documentation, https://github.com/daizutabi/hydraflow
+Project-URL: Documentation, https://daizutabi.github.io/hydraflow/
 Project-URL: Source, https://github.com/daizutabi/hydraflow
 Project-URL: Issues, https://github.com/daizutabi/hydraflow/issues
 Author-email: daizutabi <daizutabi@gmail.com>

{hydraflow-0.5.2 → hydraflow-0.5.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "hydraflow"
-version = "0.5.2"
+version = "0.5.3"
 description = "Hydraflow integrates Hydra and MLflow to manage and track machine learning experiments."
 readme = "README.md"
 license = { file = "LICENSE" }
@@ -22,7 +22,7 @@ requires-python = ">=3.10"
 dependencies = ["hydra-core>=1.3", "mlflow>=2.15"]
 
 [project.urls]
-Documentation = "https://github.com/daizutabi/hydraflow"
+Documentation = "https://daizutabi.github.io/hydraflow/"
 Source = "https://github.com/daizutabi/hydraflow"
 Issues = "https://github.com/daizutabi/hydraflow/issues"
 

{hydraflow-0.5.2 → hydraflow-0.5.3}/src/hydraflow/mlflow.py

@@ -214,7 +214,7 @@ def _list_runs(
             loc = experiment.artifact_location
 
             if isinstance(loc, str):
-                if loc.startswith("file://"):
+                if loc.startswith("file:"):
                     path = Path(mlflow.artifacts.download_artifacts(loc))
                 elif Path(loc).is_dir():
                     path = Path(loc)

{hydraflow-0.5.2 → hydraflow-0.5.3}/src/hydraflow/utils.py

@@ -33,7 +33,7 @@ def get_artifact_dir(run: Run | None = None) -> Path:
     if not isinstance(uri, str):
         raise NotImplementedError
 
-    if uri.startswith("file://"):
+    if uri.startswith("file:"):
         return Path(mlflow.artifacts.download_artifacts(uri))
 
     if Path(uri).is_dir():

{hydraflow-0.5.2 → hydraflow-0.5.3}/tests/test_mlflow.py

@@ -1,15 +1,9 @@
-import sys
 from pathlib import Path
 
 import mlflow
 import pytest
 from mlflow.entities import Experiment, Run, RunStatus
 
-pytestmark = pytest.mark.skipif(
-    sys.platform == "win32",
-    reason="Windows is not supported",
-)
-
 
 @pytest.fixture(scope="module")
 def experiment(experiment_name: str):
@@ -36,8 +30,13 @@ def test_set_experiment_uri(experiment: Experiment):
 
 
 def test_set_experiment_location(experiment: Experiment):
-    loc = Path.cwd() / "mlruns" / experiment.experiment_id
-    assert Path(experiment.artifact_location) == loc
+    loc = experiment.artifact_location
+    assert isinstance(loc, str)
+    if loc.startswith("file:"):  # for windows
+        loc = loc[loc.index("C:") :]
+
+    path = Path.cwd() / "mlruns" / experiment.experiment_id
+    assert path == Path(loc)
 
 
 def test_set_experiment_name(experiment: Experiment):
@@ -68,8 +67,12 @@ def test_log_params(run: Run, experiment_name):
 def test_get_artifact_dir_from_utils(run: Run, experiment: Experiment):
     from hydraflow.utils import get_artifact_dir
 
-    path = Path(experiment.artifact_location) / run.info.run_id / "artifacts"
-    assert get_artifact_dir(run) == path
+    loc = experiment.artifact_location
+    assert isinstance(loc, str)
+    if loc.startswith("file:"):  # for windows
+        loc = loc[loc.index("C:") :]
+
+    assert get_artifact_dir(run) == Path(loc) / run.info.run_id / "artifacts"
 
 
 @pytest.mark.parametrize(