Fix handler name

SJTU-DMTai · Mar 31, 2024 · 648ab75 · 648ab75
1 parent 6f9d051
commit 648ab75
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/benchmarks/MASTER/main.py b/examples/benchmarks/MASTER/main.py
@@ -41,8 +41,8 @@ def parse_args():
         config = yaml.safe_load(f)
 
     h_conf = config["task"]["dataset"]["kwargs"]["handler"]
-    h_path = DIRNAME / f'handler_{config["task"]["dataset"]["kwargs"]["segments"]["train"]}' \
-                       f'_{config["task"]["dataset"]["kwargs"]["segments"]["test"]}.pkl'
+    h_path = DIRNAME / f'handler_{config["task"]["dataset"]["kwargs"]["segments"]["train"][0].strftime("%Y%m%d")}' \
+                       f'_{config["task"]["dataset"]["kwargs"]["segments"]["test"][1].strftime("%Y%m%d")}.pkl'
     if not h_path.exists():
         h = init_instance_by_config(h_conf)
         h.to_pickle(h_path, dump_all=True)