refactor the code in 'ml/data_loaders'

tdspora · Jan 3, 2025 · e7ea7d9 · e7ea7d9
1 parent 3a7c970
commit e7ea7d9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/syngen/VERSION b/src/syngen/VERSION
@@ -1 +1 @@
-0.10.2rc1
+0.10.2rc2
diff --git a/src/syngen/ml/data_loaders/data_loaders.py b/src/syngen/ml/data_loaders/data_loaders.py
@@ -302,7 +302,7 @@ def load_data(self, **kwargs) -> Tuple[pd.DataFrame, Dict]:
         try:
             df = self._load_data()
             schema = self.load_schema()
-            return self._preprocess_schema_and_df(schema, df)
+            return self._get_schema_and_df(schema, df)
         except FileNotFoundError as error:
             message = (
                 f"It seems that the path to the table isn't valid.\n"
@@ -351,7 +351,7 @@ def load_schema(self) -> Dict[str, str]:
         return self._get_preprocessed_schema(original_schema)
 
     @staticmethod
-    def _preprocess_schema_and_df(
+    def _get_schema_and_df(
         schema: Dict[str, str], df: pd.DataFrame
     ) -> Tuple[pd.DataFrame, Dict[str, str]]:
         """