fix a sample data bug

OuyangWenyu · Apr 10, 2023 · b1ac6e0 · b1ac6e0
1 parent 0f85557
commit b1ac6e0
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/hydromodel/app/calibrate_xaj.py b/hydromodel/app/calibrate_xaj.py
@@ -34,10 +34,11 @@ def calibrate(args):
     algo_info = args.algorithm
     comment = args.comment
     data_dir = os.path.join(definitions.ROOT_DIR, "hydromodel", "example", exp)
-    kfold = []
-    for f_name in os.listdir(data_dir):
-        if fnmatch.fnmatch(f_name, "*_fold*_test.json"):
-            kfold.append(int(f_name[len("data_info_fold") : -len("_test.json")]))
+    kfold = [
+        int(f_name[len("data_info_fold") : -len("_test.json")])
+        for f_name in os.listdir(data_dir)
+        if fnmatch.fnmatch(f_name, "*_fold*_test.json")
+    ]
     kfold = np.sort(kfold)
     for fold in kfold:
         print(f"Start to calibrate the {fold}-th fold")