fix read_datastore

src/data.py

@@ -91,16 +91,13 @@ def format_area(value):
     print(f"Refactored data saved to {data_path}")
 
 
-def read_datastore():
-    with initialize(config_path="../configs", version_base=None):
-        cfg = compose(config_name="main")
-
+@hydra.main(config_path="../configs", config_name="main", version_base=None)
+def read_datastore(cfg: DictConfig):
     version = cfg.test_data_version if cfg.test else cfg.index
     try:
-        subprocess.run(["dvc", "fetch"])
+        subprocess.run(["dvc", "fetch", f"{cfg.dvc_file_path}"])
         subprocess.run(["dvc", "pull"], check=True)
-        subprocess.run(
-            ["git", "checkout", f"v{version}.0", f"{cfg.dvc_file_path}"], check=True)
+        subprocess.run(["git", "checkout", f"v{version}.0", f"{cfg.dvc_file_path}"], check=True)
         subprocess.run(["dvc", "checkout", f"{cfg.dvc_file_path}"], check=True)
 
         sample_path = cfg.sample_path
@@ -110,8 +107,7 @@ def read_datastore():
         return sample
     finally:
         # Return to the HEAD state
-        subprocess.run(["git", "checkout", "HEAD",
-                       f"{cfg.dvc_file_path}"], check=True)
+        subprocess.run(["git", "checkout", "HEAD", f"{cfg.dvc_file_path}"], check=True)
         subprocess.run(["dvc", "checkout", f"{cfg.dvc_file_path}"], check=True)