'load_task' is not defined

[zdx3578]

snap-stanford/ViRel#1

---

NameError Traceback (most recent call last)
Cell In[23], line 6
4 if filename.endswith(".json"):
5 print("extracting objs from task: ", filename)
----> 6 training_objs.extend(single_task_obj_parser(filename.split(".")[0]))
7 # we only need to parse once, and save it to the disk
8 torch.save(training_objs, os.path.join(tmp_data_dirname, 'arc_objs.pt'))

File ~/github/VSAHDC/virel/BabyARC/code/dataset/utils.py:223, in single_task_obj_parser(task_id)
221 task_objs = []
222 task_id += ".json"
--> 223 (inputs, ), (, _) = load_task(task_id, isplot=False)
224 # parse objects in 3 ways
225 inputs_graph = parse_obj(inputs)

NameError: name 'load_task' is not defined