Warning fix

sdv-dev · Aug 16, 2023 · 095e73b · 095e73b
1 parent 6373a47
commit 095e73b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ctgan/data_transformer.py b/ctgan/data_transformer.py
@@ -46,7 +46,7 @@ def _fit_continuous(self, data):
                 A ``ColumnTransformInfo`` object.
         """
         column_name = data.columns[0]
-        gm = ClusterBasedNormalizer(model_missing_values=True, max_clusters=min(len(data), 10))
+        gm = ClusterBasedNormalizer(max_clusters=min(len(data), 10))
         gm.fit(data, column_name)
         num_components = sum(gm.valid_component_indicator)
 

diff --git a/tests/unit/test_data_transformer.py b/tests/unit/test_data_transformer.py
@@ -75,7 +75,7 @@ def test__fit_continuous_max_clusters(self, MockCBN):
         transformer._fit_continuous(data)
 
         # Assert
-        MockCBN.assert_called_once_with(model_missing_values=True, max_clusters=len(data))
+        MockCBN.assert_called_once_with(max_clusters=len(data))
 
     @patch('ctgan.data_transformer.OneHotEncoder')
     def test___fit_discrete(self, MockOHE):