chore: fix linting error.

ydataai · Jan 29, 2025 · ed086c7 · ed086c7
1 parent 0f319d3
commit ed086c7
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 3 deletions.
diff --git a/src/ydata/sdk/datasources/datasource.py b/src/ydata/sdk/datasources/datasource.py
@@ -128,7 +128,6 @@ def get(uid: UID, project: Optional[Project] = None, client: Optional[Client] =
         datasource_type = CONNECTOR_TO_DATASOURCE.get(
             ConnectorType(data['connector']['type']))
         datasource = DataSource._model_from_api(data, datasource_type)
-        #datasource = DataSource._init_from_model_data(model)
         datasource._project = project
         return datasource
 

diff --git a/src/ydata/sdk/synthesizers/multitable.py b/src/ydata/sdk/synthesizers/multitable.py
@@ -53,7 +53,6 @@ def fit(self, X,
         Arguments:
             X (DataSource): DataSource to Train
         """
-        #self._fit_from_datasource(X, datatype=DataSourceType.MULTITABLE)
         super().fit(X, datatype=DataSourceType.MULTITABLE)
 
     def sample(self, frac: Union[int, float] = 1, write_connector: Optional[Union[Connector, UID]] = None) -> None:

diff --git a/src/ydata/sdk/synthesizers/synthesizer.py b/src/ydata/sdk/synthesizers/synthesizer.py
@@ -172,7 +172,7 @@ def _validate_datasource_attributes(X, dataset_attrs: DataSourceAttrs, datatype:
                 raise DataTypeMissingError(
                     "Argument `datatype` is mandatory for pandas.DataFrame training data")
         elif datatype == DataSourceType.MULTITABLE:
-            tables = [t for t in X.tables.keys()]
+            tables = [t for t in X.tables.keys()] # noqa: F841
             # Does it make sense to add more validations here?
         else:
             columns = [c.name for c in X.metadata.columns]