Update models/spabert/datasets/osm_sample_loader.py

models/spabert/datasets/osm_sample_loader.py

@@ -8,8 +8,9 @@ import torch
 from transformers import RobertaTokenizer, BertTokenizer
 from torch.utils.data import Dataset
 #sys.path.append('/home/zekun/spatial_bert/spatial_bert/datasets')
-sys.path.append('/content/drive/MyDrive/spaBERT/spabert/datasets')
-from dataset_loader_ver2 import SpatialDataset
+#sys.path.append('/content/drive/MyDrive/spaBERT/spabert/datasets')
+from models.spabert.datasets.dataset_loader_ver2 import SpatialDataset
+#from dataset_loader_ver2 import SpatialDataset
 
 import pdb