fixing bug in folder names

IntelCompH2020 · Oct 18, 2023 · ed65a1e · ed65a1e
1 parent 1ef3ef2
commit ed65a1e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/data_manager.py b/src/data_manager.py
@@ -628,8 +628,8 @@ def sample_sub_set_from_folder(path2texts, selected_cols,
                        'appln_abstract': 'description'}
 
             path2texts = self.path2corpus / 'corpus'
-            path2_c1 = path2texts / 'patstat_Kwds3_AI.parquet'
-            path2_c0 = path2texts / 'patstat_Kwds_AI.parquet'
+            path2_c1 = path2texts / 'PATSTAT_Kwds3_AI.parquet'
+            path2_c0 = path2texts / 'PATSTAT_Kwds_AI.parquet'
 
             df_c1 = self._load_parquet(path2_c1, columns, mapping)
             df_corpus = self._load_parquet(path2_c0, columns, mapping)