changes after review

tdspora · Oct 29, 2024 · 2980e2e · 2980e2e
1 parent 3d265cb
commit 2980e2e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/src/syngen/ml/handlers/handlers.py b/src/syngen/ml/handlers/handlers.py
@@ -500,10 +500,15 @@ def handle(self, **kwargs):
             prepared_data = prepared_data.drop(tech_columns, axis=1)
             logger.debug(
                 f"Technical columns "
-                f"{self.dataset.tech_columns} were dropped"
+                f"{tech_columns} were dropped"
                 f" from the generated table"
                 )
-            Report().unregister_reporters(self.dataset.table_name)
+            Report().unregister_reporters(self.table_name)
+            logger.info(
+                    "Since there were no columns to train on, "
+                    "reports will be not generated "
+                    f"for the table '{self.table_name}'."
+                    )
 
         is_pk = self._is_pk()
 

diff --git a/src/syngen/ml/reporters/reporters.py b/src/syngen/ml/reporters/reporters.py
@@ -218,12 +218,6 @@ def unregister_reporters(cls, table: str):
         """
         cls._reporters[table] = list()
 
-        logger.info(
-                "Since there were no columns to train on, "
-                "reports will be not generated "
-                f"for the table '{table}'."
-                )
-
     @classmethod
     def clear_report(cls):
         """