ImportError: cannot import name 'SpatialDataset' from 'dataset_loader'

[mahrokh3409]

Dear @zekun-li ,

I am attempting to run the following command for pre-training:
python3 train_joint.py --model_save_dir=OUTPUT_WEIGHT_DIR --pseudo_sentence_dir='../../datasets/osm_pseudo_sent/world/' --nl_sentence_dir='../../datasets/wikidata/world_georelation/joint_v2/' --batch_size=28 --lr=1e-5 --spatial_dist_fill=900 --placename_to_osmid_path='../../datasets/osm_pseudo_sent/name-osmid-dict/placename_to_osmid.json'

However, I encountered the following error:

ImportError: cannot import name 'SpatialDataset' from 'dataset_loader'

Thank you for your assistance.

Kind Regards,
Mah