Fixed import

malteos · Feb 25, 2022 · 145a69f · 145a69f
1 parent 25c4976
commit 145a69f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cli_specter.py b/cli_specter.py
@@ -54,7 +54,7 @@ def extract_triples(
 
     logger.info(f'Parsing {pickle_path}')
 
-    from gdt.datasets.specter_dataset import DataReaderFromPickled
+    from gdt.datasets.specter import DataReaderFromPickled
 
     datareaderfp = DataReaderFromPickled(max_sequence_length=512)
     data_instances = datareaderfp._read(pickle_path)