Fix a bug

SJTU-DMTai · Mar 30, 2024 · c32dbc8 · c32dbc8
1 parent c77520c
commit c32dbc8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/benchmarks/MASTER/main.py b/examples/benchmarks/MASTER/main.py
@@ -46,7 +46,7 @@ def parse_args():
         h = init_instance_by_config(h_conf)
         h.to_pickle(h_path, dump_all=True)
         print('Save preprocessed data to', h_path)
-    config["dataset"]["kwargs"]["handler"] = f"file://{h_path}"
+    config["task"]["dataset"]["kwargs"]["handler"] = f"file://{h_path}"
     dataset = init_instance_by_config(config['task']["dataset"])
 
     ###################################