Resolve merge conflict with config branch

geoelements · Jun 28, 2024 · e3283cf · e3283cf
2 parents e28275c + a28603e
commit e3283cf
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/.github/workflows/train.yml b/.github/workflows/train.yml
@@ -12,6 +12,10 @@ jobs:
     steps:
     - name: Checkout repository
       uses: actions/checkout@v4
+
+    - name: Black linter
+      run: |
+        black --check .
 
     - name: PyTest
       run: |
@@ -22,8 +26,4 @@ jobs:
         TMP_DIR="../gns-sample"
         DATASET_NAME="WaterDropSample"
         git clone https://github.com/geoelements/gns-sample ../gns-sample
-        python -m gns.train mode="train" training.steps=10
-
-    - name: Black linter
-      run: |
-        black --check .
+        python -m gns.train mode="train" training.steps=10
diff --git a/gns/train.py b/gns/train.py
@@ -139,7 +139,7 @@ def predict(device: str, cfg: DictConfig):
     # Use `valid`` set for eval mode if not use `test`
     split = (
         "test"
-        if (cfg.mode == "rollout" or (not os.path.isfile("{cfg.data_path}valid.npz")))
+        if (cfg.mode == "rollout" or (not os.path.isfile("{cfg.data.path}valid.npz")))
         else "valid"
     )