hydraflow 0.7.2__py3-none-any.whl → 0.7.4__py3-none-any.whl

hydraflow/cli.py

@@ -0,0 +1,75 @@
+"""Hydraflow CLI."""
+
+from __future__ import annotations
+
+from pathlib import Path
+from typing import Annotated
+
+import typer
+from omegaconf import DictConfig, OmegaConf
+from rich.console import Console
+from typer import Argument, Option
+
+app = typer.Typer(add_completion=False)
+console = Console()
+
+
+@app.command()
+def run(
+    names: Annotated[
+        list[str] | None,
+        Argument(help="Job names.", show_default=False),
+    ] = None,
+) -> None:
+    """Run jobs."""
+    typer.echo(names)
+
+    cfg = load_config()
+    typer.echo(cfg)
+
+
+@app.command()
+def show() -> None:
+    """Show the config."""
+    from rich.syntax import Syntax
+
+    cfg = load_config()
+    code = OmegaConf.to_yaml(cfg)
+    syntax = Syntax(code, "yaml")
+    console.print(syntax)
+
+
+@app.callback(invoke_without_command=True)
+def callback(
+    *,
+    version: Annotated[
+        bool,
+        Option("--version", help="Show the version and exit."),
+    ] = False,
+) -> None:
+    if version:
+        import importlib.metadata
+
+        typer.echo(f"hydraflow {importlib.metadata.version('hydraflow')}")
+        raise typer.Exit
+
+
+def find_config() -> Path:
+    if Path("hydraflow.yaml").exists():
+        return Path("hydraflow.yaml")
+
+    if Path("hydraflow.yml").exists():
+        return Path("hydraflow.yml")
+
+    typer.echo("No config file found.")
+    raise typer.Exit(code=1)
+
+
+def load_config() -> DictConfig:
+    cfg = OmegaConf.load(find_config())
+
+    if isinstance(cfg, DictConfig):
+        return cfg
+
+    typer.echo("Invalid config file.")
+    raise typer.Exit(code=1)

hydraflow/main.py

@@ -23,10 +23,9 @@ def main(
     node: Any,
     config_name: str = "config",
     *,
-    chdir: bool = True,
-    skip_finished: bool = True,
+    chdir: bool = False,
     force_new_run: bool = False,
-    override: bool = True,
+    skip_finished: bool = True,
 ):
     """Main decorator."""
 
@@ -42,7 +41,7 @@ def main(
                 run = None
             else:
                 rc = hydraflow.search_runs()
-                run = rc.try_get(cfg, override=override)
+                run = rc.try_get(cfg, override=True)
 
                 if skip_finished and run and run.info.status == FINISHED:
                     return

{hydraflow-0.7.2.dist-info → hydraflow-0.7.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hydraflow
-Version: 0.7.2
+Version: 0.7.4
 Summary: Hydraflow integrates Hydra and MLflow to manage and track machine learning experiments.
 Project-URL: Documentation, https://daizutabi.github.io/hydraflow/
 Project-URL: Source, https://github.com/daizutabi/hydraflow
@@ -38,6 +38,9 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.10
 Requires-Dist: hydra-core>=1.3
 Requires-Dist: mlflow>=2.15
+Requires-Dist: omegaconf
+Requires-Dist: rich
+Requires-Dist: typer
 Description-Content-Type: text/markdown
 
 # Hydraflow

{hydraflow-0.7.2.dist-info → hydraflow-0.7.4.dist-info}/RECORD

@@ -1,7 +1,8 @@
 hydraflow/__init__.py,sha256=rujOGabEPPhPfyqTHynem3unqIEQ1haTWWSMuu2LuoQ,898
+hydraflow/cli.py,sha256=jxqFppNeJWAr2Tb-C_MQXEJtegJ6TXcd3C1CT7Jdb1A,1559
 hydraflow/config.py,sha256=MNX9da5bPVDcjnpji7Cm9ndK6ura92pt361m4PRh6_E,4326
 hydraflow/context.py,sha256=3xfKhMozkKFqtWeOp9Gie0A5o5URMta4US6iVD5TcLU,6002
-hydraflow/main.py,sha256=8DYYxyyS1sToEIoIQlzGcGDNgzIFg3fLnjBufQtQASc,1344
+hydraflow/main.py,sha256=hroncI_SNpNgEtdxLgzI397J5S2Amv7J0atnPxwBePM,1314
 hydraflow/mlflow.py,sha256=imD3XL0RTlpnKrkyvO8FNy_Bv6hwSfLiOu1yJuL40ck,8773
 hydraflow/param.py,sha256=yu1aMNXRLegXGDL-68vwIkfeDF9CaU784WZENGLwl7Q,4572
 hydraflow/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -9,7 +10,8 @@ hydraflow/run_collection.py,sha256=YCWg5Dz1j49xB2LA75onq5wsAeQQbifXpG4yPUwRN4I,2
 hydraflow/run_data.py,sha256=dpyyfnuH9mCtIZeigMo1iFQo9bafMdEL4i4uI2l0UqY,1525
 hydraflow/run_info.py,sha256=Jf5wrIjRLIV1-k-obHDqwKHa6j_ZonrY8od-rXlbtMo,1024
 hydraflow/utils.py,sha256=a9i5PEJn8Ssowv9dqHadAihZXlsqtVjHZ9MZvkPq1bY,4747
-hydraflow-0.7.2.dist-info/METADATA,sha256=TJfrv_OjNiZskkog8UFmJI9wXjatNBKFTovPdMjnbK8,4700
-hydraflow-0.7.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hydraflow-0.7.2.dist-info/licenses/LICENSE,sha256=IGdDrBPqz1O0v_UwCW-NJlbX9Hy9b3uJ11t28y2srmY,1062
-hydraflow-0.7.2.dist-info/RECORD,,
+hydraflow-0.7.4.dist-info/METADATA,sha256=GTJi5z8TTIwPy6qpscw-t3Mb1V-GOR0iYU_IB-DB-UE,4766
+hydraflow-0.7.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hydraflow-0.7.4.dist-info/entry_points.txt,sha256=XI0khPbpCIUo9UPqkNEpgh-kqK3Jy8T7L2VCWOdkbSM,48
+hydraflow-0.7.4.dist-info/licenses/LICENSE,sha256=IGdDrBPqz1O0v_UwCW-NJlbX9Hy9b3uJ11t28y2srmY,1062
+hydraflow-0.7.4.dist-info/RECORD,,

hydraflow-0.7.4.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+hydraflow = hydraflow.cli:app