Remove torch and numpy

marvinsxtr · marvinsxtr · commit 940c0970a64f · 2025-09-12T00:02:49.000+02:00
diff --git a/LICENSE b/LICENSE
@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2024 Marvin Sextro
+Copyright (c) 2025 Marvin Sextro
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal
diff --git a/example/configs.py b/example/configs.py
@@ -3,7 +3,7 @@
 from ml_project_template.runs import Job, Run, SlurmParams, SweepJob
 from ml_project_template.wandb import WandBRun
 
-RunConfig = builds(Run, seed=42, wandb=None, job=None)
+RunConfig = builds(Run, seed=None, wandb=None, job=None)
 
 SlurmParamsConfig = builds(
     SlurmParams,
@@ -18,6 +18,6 @@
 
 JobConfig = builds(Job, slurm_params=SlurmParamsConfig)
 
-SweepConfig = builds(SweepJob, num_workers=2, parameters={"cfg.seed": [42, 1337]}, builds_bases=(JobConfig,))
+SweepConfig = builds(SweepJob, num_workers=2, parameters={"foo": [42, 1337]}, builds_bases=(JobConfig,))
 
 WandBConfig = builds(WandBRun, group=None, mode="online")
diff --git a/ml_project_template/config.py b/ml_project_template/config.py
@@ -7,15 +7,16 @@
 from hydra_zen.third_party.pydantic import pydantic_parser
 from omegaconf import DictConfig, OmegaConf
 
-from ml_project_template.utils import ConfigKeys, get_hydra_output_dir, logger, seed_everything
+from ml_project_template.utils import ConfigKeys, get_output_dir, logger
 from ml_project_template.wandb import WandBRun
 
 
-def pre_call(root_config: DictConfig, log_debug: bool = False) -> None:
+def pre_call(root_config: DictConfig, seed_fn: Callable[[int], None] | None = None, log_debug: bool = False) -> None:
     """Logs the config, sets the seed and initializes a WandB run before config instantiation.
 
     Args:
         root_config: Unresolved config.
+        seed_fn: Function to use for seeding the run.
         log_debug: Whether to log the config, seed and output path.
     """
     if log_debug:
@@ -27,7 +28,10 @@ def pre_call(root_config: DictConfig, log_debug: bool = False) -> None:
         return
 
     if (seed := config.get(ConfigKeys.SEED)) is not None:
-        seed_everything(seed)
+        if seed_fn is None:
+            raise ValueError("No seeding function was set for the given seed.")
+
+        seed_fn(seed)
         logger.debug(f"Set seed to {seed}.")
     else:
         logger.warning("No seed was configured! Run may not be reproducible.")
@@ -37,7 +41,7 @@ def pre_call(root_config: DictConfig, log_debug: bool = False) -> None:
     else:
         logger.debug(f"Running config:\n{to_yaml(root_config)}")
 
-    output_path = get_hydra_output_dir()
+    output_path = get_output_dir()
     logger.debug(f"Saving outputs in {output_path}")
 
     logger.setLevel(logging.INFO)
@@ -48,17 +52,18 @@ def pre_call(root_config: DictConfig, log_debug: bool = False) -> None:
         wandb.save(output_path / ".hydra/*", base_path=output_path, policy="now")
 
 
-def run(main_function: Callable, log_debug: bool = True) -> None:
+def run(main_function: Callable, seed_fn: Callable[[int], None] | None = None, log_debug: bool = True) -> None:
     """Configure and run a given function using hydra-zen.
 
     Args:
         main_function: Function to configure and run.
+        seed_fn: Function to use for seeding the run.
         log_debug: Whether to log debug information from the `pre_call` function.
     """
     store.add_to_hydra_store()
     zen(
         main_function,
-        pre_call=partial(pre_call, log_debug=log_debug),
+        pre_call=partial(pre_call, seed_fn=seed_fn, log_debug=log_debug),
         resolve_pre_call=False,
         instantiation_wrapper=pydantic_parser,
     ).hydra_main(
diff --git a/ml_project_template/runs.py b/ml_project_template/runs.py
@@ -9,7 +9,7 @@
 from submitit import AutoExecutor
 from submitit.helpers import CommandFunction
 
-from ml_project_template.utils import ConfigKeys, get_hydra_output_dir, logger
+from ml_project_template.utils import ConfigKeys, get_output_dir, logger
 from ml_project_template.wandb import WandBConfig, WandBRun
 
 
@@ -80,7 +80,7 @@ def run(self) -> None:
 
         function = CommandFunction(command)
         executor = AutoExecutor(
-            folder=get_hydra_output_dir(),
+            folder=get_output_dir(),
             cluster=self.cluster,
             slurm_python=self.python_command,
         )
@@ -134,7 +134,7 @@ def run(self) -> None:
         metric = {"goal": self.metric_goal, "name": self.metric_name}
         program, args = sys.argv[0], self.filter_args(sys.argv[1:])
 
-        folder_path = get_hydra_output_dir()
+        folder_path = get_output_dir()
         dummy_sweep_id = "sweep_started_" + Path(folder_path).parts[-2] + "_" + Path(folder_path).parts[-1]
         hydra_run_dir = "./outputs/sweeps/" + dummy_sweep_id + "/${now:%H-%M-%S-%f}"
 
@@ -179,6 +179,6 @@ def run(self) -> None:
 class Run:
     """Configures a basic run."""
 
-    seed: int
+    seed: int | None = None
     wandb: WandBRun | None = None
     job: Job | None = None
diff --git a/ml_project_template/utils.py b/ml_project_template/utils.py
@@ -4,8 +4,6 @@
 from pathlib import Path
 from typing import Final
 
-import numpy as np
-import torch
 from hydra.core.hydra_config import HydraConfig
 
 logger = logging.getLogger()
@@ -21,37 +19,25 @@ class ConfigKeys:
     STORE: Final[str] = "store"
 
 
-def seed_everything(seed: int) -> None:
-    """Seeds all random number generators.
+def basic_seed_fn(seed: int) -> None:
+    """Seeds random number generators.
 
     Args:
         seed: Random seed.
     """
     random.seed(seed)
-    np.random.seed(seed)
-    torch.manual_seed(seed)
-    torch.cuda.manual_seed_all(seed)
     os.environ["PYTHONHASHSEED"] = str(seed)
 
 
-def get_device() -> str:
-    """Returns the available device for torch.
+def get_output_dir() -> Path:
+    """Get the current output directory.
 
     Returns:
-        The GPU or the MPS device when available and the CPU device as a fallback.
+    Output path of the current run.
     """
-    if torch.cuda.is_available():
-        return "cuda"
-    elif torch.backends.mps.is_available():
-        return "mps"
-    else:
-        return "cpu"
-
-
-def get_hydra_output_dir() -> Path:
-    """Return the hydra output directory.
-
-    Returns:
-        Path to the hydra output directory.
-    """
-    return Path(HydraConfig.get().runtime.output_dir)
+    try:
+        output_dir = Path(HydraConfig.get().runtime.output_dir)
+    except ValueError:
+        output_dir = Path("/tmp/outputs")
+        output_dir.mkdir(exist_ok=True, parents=True)
+    return output_dir
diff --git a/ml_project_template/wandb.py b/ml_project_template/wandb.py
@@ -3,7 +3,7 @@
 from typing import Self
 
 import wandb
-from dotenv import load_dotenv
+from dotenv import find_dotenv, load_dotenv
 from wandb.wandb_run import Run
 
 from ml_project_template.utils import logger
@@ -25,7 +25,7 @@ def from_env(cls) -> Self | None:
             Populated `WandBConfig` or None if environment variables could not be found.
         """
         config = None
-        load_dotenv()
+        load_dotenv(find_dotenv(usecwd=True))
 
         try:
             config = cls(**{field.name: os.environ[field.name] for field in fields(cls)})
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,51 +4,31 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ml_project_template"
-version = "0.0.12"
+version = "0.0.14"
 requires-python = ">=3.12"
 dependencies = [
-  "torch>=2.5.1",
   "wandb>=0.18.5",
   "hydra-zen>=0.14.0",
   "submitit>=1.5.1",
   "python-dotenv>=1.0.1",
-  "numpy>=1.26.4",
   "pydantic>=2.10.3",
   "pytest>=8.3.5",
 ]
 
 [dependency-groups]
-dev = [
-    "pre-commit>=4.0.1",
-    "ruff>=0.8.0",
-]
+dev = ["pre-commit>=4.0.1", "ruff>=0.8.0"]
 
 [tool.setuptools.packages.find]
 include = ["ml_project_template"]
 namespaces = true
 
-[tool.uv.sources]
-torch = [
-  { index = "pytorch-cu124", marker = "platform_system != 'Darwin'" },
-]
-
-[[tool.uv.index]]
-name = "pytorch-cu124"
-url = "https://download.pytorch.org/whl/cu124"
-explicit = true
-
 [tool.pytest.ini_options]
 pythonpath = ["."]
 
 [tool.mypy]
 explicit_package_bases = true
 disable_error_code = ["import-untyped"]
 
-[[tool.mypy.overrides]]
-module = "cloudpathlib.*"
-ignore_errors = true
-follow_imports = "skip"
-
 [tool.ruff]
 line-length = 119
 indent-width = 4
@@ -59,67 +39,63 @@ convention = "google"
 [tool.ruff.lint]
 select = ["ALL"]
 ignore = [
-    "ANN002",
-    "ANN003",
-    "ANN401",
-    "D413",
-    "COM812",
-    "D100",
-    "D104",
-    "D107",
-    "D205",
-    "PD901",
-    "D400",
-    "D401",
-    "D415",
-    "FA",
-    "SLF",
-    "INP",
-    "TRY003",
-    "TRY201",
-    "EM",
-    "FBT",
-    "RET",
-    "C406",
-    "E741",
-    "PLR2004",
-    "RUF009",
-    "RUF012",
-    "BLE001",
-    "S603",
-    "S607",
-    "S506",
-    "FIX002",
-    "NPY002",
-    "G004",
-    "S311",
-    "PIE790",
-    "TRY400",
-    "S108",
-    "W191",
-    "E111",
-    "E114",
-    "E117",
-    "D206",
-    "D300",
-    "Q000",
-    "Q001",
-    "Q002",
-    "Q003",
-    "COM812",
-    "COM819",
-    "D203",
-    "D213",
-    "N806",
-    "N803",
-    "E712",
-    "PLR0913",
-    "TC001"
+  "ANN002",
+  "ANN003",
+  "ANN401",
+  "D413",
+  "COM812",
+  "D100",
+  "D104",
+  "D107",
+  "D205",
+  "PD901",
+  "D400",
+  "D401",
+  "D415",
+  "FA",
+  "SLF",
+  "INP",
+  "TRY003",
+  "TRY201",
+  "EM",
+  "FBT",
+  "RET",
+  "C406",
+  "E741",
+  "PLR2004",
+  "RUF009",
+  "RUF012",
+  "BLE001",
+  "S603",
+  "S607",
+  "S506",
+  "FIX002",
+  "NPY002",
+  "G004",
+  "S311",
+  "PIE790",
+  "TRY400",
+  "S108",
+  "W191",
+  "E111",
+  "E114",
+  "E117",
+  "D206",
+  "D300",
+  "Q000",
+  "Q001",
+  "Q002",
+  "Q003",
+  "COM812",
+  "COM819",
+  "D203",
+  "D213",
+  "N806",
+  "N803",
+  "E712",
+  "PLR0913",
+  "TC001",
 ]
 
 [tool.ruff.lint.per-file-ignores]
-"**/tests/**/*.py" = [
-    "S101",
-    "ARG",
-    "FBT",
-]
+"**/tests/**/*.py" = ["S101", "ARG", "FBT"]
diff --git a/uv.lock b/uv.lock
