loading dataset!
[rank0]: Traceback (most recent call last):
[rank0]:   File "/mnt/data/ZeroEval/src/unified_infer.py", line 152, in <module>
[rank0]:     id_strs, chat_history, model_inputs, metadata = load_eval_data(args)
[rank0]:   File "/mnt/data/ZeroEval/src/unified_utils.py", line 69, in load_eval_data
[rank0]:     dataset, id_name = mapping_task_names(data_name)
[rank0]:   File "/mnt/data/ZeroEval/src/task_configs.py", line 14, in mapping_task_names
[rank0]:     dataset = load_dataset("allenai/ZebraLogicBench", "grid_mode", split="test")
[rank0]:   File "/usr/local/lib/python3.10/dist-packages/datasets/load.py", line 2606, in load_dataset
[rank0]:     builder_instance = load_dataset_builder(
[rank0]:   File "/usr/local/lib/python3.10/dist-packages/datasets/load.py", line 2277, in load_dataset_builder
[rank0]:     dataset_module = dataset_module_factory(
[rank0]:   File "/usr/local/lib/python3.10/dist-packages/datasets/load.py", line 1923, in dataset_module_factory
[rank0]:     raise e1 from None
[rank0]:   File "/usr/local/lib/python3.10/dist-packages/datasets/load.py", line 1843, in dataset_module_factory
[rank0]:     dataset_info = hf_api.dataset_info(
[rank0]:   File "/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn
[rank0]:     return fn(*args, **kwargs)
[rank0]:   File "/usr/local/lib/python3.10/dist-packages/huggingface_hub/hf_api.py", line 2366, in dataset_info
[rank0]:     return DatasetInfo(**data)
[rank0]:   File "/usr/local/lib/python3.10/dist-packages/huggingface_hub/hf_api.py", line 799, in __init__
[rank0]:     self.tags = kwargs.pop("tags")
[rank0]: KeyError: 'tags'
INFO 01-09 16:51:48 model_runner.py:1456] Graph capturing finished in 17 secs.