linting

Author: andreatgretel

src/data_designer/engine/column_generators/generators/custom.py

@@ -13,7 +13,6 @@
 )
 from data_designer.engine.errors import DataDesignerRuntimeError
 
-
 logger = logging.getLogger(__name__)
 
 
@@ -51,4 +50,4 @@ def generate(self, data: pd.DataFrame) -> pd.DataFrame:
         except Exception as e:
             raise DataDesignerRuntimeError(f"Error generating custom column {self.config.name!r}: {e}")
 
-        return result
+        return result

src/data_designer/essentials/__init__.py

@@ -80,6 +80,7 @@
     "CodeValidatorParams",
     "ColumnInequalityConstraint",
     "configure_logging",
+    "CustomColumnConfig",
     "DataDesignerColumnType",
     "DataDesignerConfig",
     "DataDesignerConfigBuilder",

tests/engine/column_generators/generators/test_custom.py

@@ -96,4 +96,4 @@ def failing_generator(data: pd.DataFrame) -> pd.DataFrame:
     with pytest.raises(
         DataDesignerRuntimeError, match=r"Error generating custom column 'test_column': Something went wrong"
     ):
-        generator.generate(df)
+        generator.generate(df)