tursom/memgraph
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
jeremy 2022-11-04 15:57:26 +01:00
parent e41073bc2c
commit 9e72c7cb54
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/mgbench/benchmark.py
View File

@ -126,7 +126,7 @@ args = parser.parse_args()
head_tail = os.path.split(args.datasets_path) head_tail = os.path.split(args.datasets_path)
path_without_dataset_name = head_tail[0] 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) sys.path.append(path_without_dataset_name)
dataset_to_use = importlib.import_module(dataset_name) dataset_to_use = importlib.import_module(dataset_name)