refactor: purge os from init files

laserkelvin · Jul 15, 2024 · cb51078 · cb51078
1 parent 8e11437
commit cb51078
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/experiments/datasets/__init__.py b/experiments/datasets/__init__.py
@@ -1,4 +1,3 @@
-import os
 import yaml
 
 from pathlib import Path
@@ -15,7 +14,7 @@
 
 
 for filename in yaml_dir.rglob("*.yaml"):
-    file_path = Path(os.path.join(yaml_dir, filename))
+    file_path = yaml_dir.joinpath(filename)
     with open(file_path, "r") as file:
         content = yaml.safe_load(file)
         file_key = file_path.stem

diff --git a/experiments/models/__init__.py b/experiments/models/__init__.py
@@ -1,4 +1,3 @@
-import os
 import yaml
 
 from torch.nn import LayerNorm
@@ -21,7 +20,7 @@
 }
 
 for filename in yaml_dir.rglob("*.yaml"):
-    file_path = Path(os.path.join(yaml_dir, filename))
+    file_path = yaml_dir.joinpath(filename)
     with open(file_path, "r") as file:
         content = yaml.safe_load(file)
         file_key = file_path.stem

diff --git a/experiments/trainer_config/__init__.py b/experiments/trainer_config/__init__.py
@@ -1,6 +1,5 @@
 from experiments.trainer_config.trainer_config import setup_trainer  # noqa: F401
 
-import os
 import yaml
 from pathlib import Path
 
@@ -11,7 +10,7 @@
 }
 
 for filename in yaml_dir.rglob("*.yaml"):
-    file_path = Path(os.path.join(yaml_dir, filename))
+    file_path = yaml_dir.joinpath(filename)
     with open(file_path, "r") as file:
         content = yaml.safe_load(file)
         file_key = file_path.stem