Add extra safe check to in case we call on dataset.py

This commit is contained in:
jeremy 2022-11-04 15:57:26 +01:00
parent e41073bc2c
commit 9e72c7cb54

View File

@ -126,7 +126,7 @@ args = parser.parse_args()
head_tail = os.path.split(args.datasets_path)
path_without_dataset_name = head_tail[0]
dataset_name = head_tail[1]
dataset_name = head_tail[1].split(".")[0]
sys.path.append(path_without_dataset_name)
dataset_to_use = importlib.import_module(dataset_name)