fix the process of providing the information of the schema depending …

…on the format of the generated data (#474) * fix the process of providing the information of the schema depending on the format of the generated data * update 'VERSION' --------- Co-authored-by: Hanna Imshenetska <[email protected]@EVZZAMZSA0021.epam.com>
tdspora · Oct 23, 2024 · b24affb · b24affb
1 parent f090210
commit b24affb
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 6 deletions.
diff --git a/src/syngen/VERSION b/src/syngen/VERSION
@@ -1 +1 @@
-0.9.45
+0.9.46
diff --git a/src/syngen/ml/data_loaders/data_loaders.py b/src/syngen/ml/data_loaders/data_loaders.py
@@ -318,6 +318,7 @@ def _save_data(self, df: pd.DataFrame, schema: Optional[Dict]):
 
     def save_data(self, df: pd.DataFrame, schema: Optional[Dict] = None, **kwargs):
         if schema is not None:
+            logger.trace(f"The data will be saved with the schema: {schema}")
             preprocessed_schema = (
                 self._get_preprocessed_schema(schema) if schema is not None else schema
             )

diff --git a/src/syngen/ml/handlers/handlers.py b/src/syngen/ml/handlers/handlers.py
@@ -502,11 +502,6 @@ def handle(self, **kwargs):
                 )
                 generated_data = generated_data[self.dataset.order_of_columns]
 
-                if self.original_schema:
-                    logger.trace(
-                        f"The synthetic data of the table - '{self.table_name}' "
-                        f"will be saved with the schema: {self.original_schema}"
-                    )
                 if generated_data is None:
                     DataLoader(self.paths["path_to_merged_infer"]).save_data(
                         prepared_data,