Update scibite_cord.py

Knowledge-Graph-Hub · Apr 10, 2020 · 4edfd94 · 4edfd94
1 parent 2e2fb44
commit 4edfd94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kg_covid_19/transform_utils/scibite_cord/scibite_cord.py b/kg_covid_19/transform_utils/scibite_cord/scibite_cord.py
@@ -63,7 +63,7 @@ def parse_annotations(self, node_handle: Any, edge_handle: Any, data_file: str)
 
         subsets = ['biorxiv_medrxiv', 'comm_use_subset', 'noncomm_use_subset', 'custom_license']
         for subset in subsets:
-            data_dir = os.path.join(self.input_base_dir, 'data', subset, subset)
+            data_dir = os.path.join(self.input_base_dir, 'CORD19', subset, subset)
             for filename in os.listdir(data_dir):
                 file = os.path.join(data_dir, filename)
                 doc = json.load(open(file))