fix broken test

adriangb · adriangb · commit 6eee3c4a19b2 · 2021-02-16T00:07:39.000-06:00
diff --git a/scikeras/utils/transformers.py b/scikeras/utils/transformers.py
@@ -419,4 +419,5 @@ def transform(self, data: Dict[str, Any]) -> Dict[str, Any]:
         sample_weight = 1 if sample_weight is None else sample_weight
         sample_weight *= compute_sample_weight(class_weight=self.class_weight, y=y)
         data["sample_weight"] = sample_weight
+        data["class_weight"] = None
         return data
diff --git a/scikeras/wrappers.py b/scikeras/wrappers.py
@@ -817,8 +817,6 @@ def _initialize(
         self.dataset_transformer_ = self.dataset_transformer.fit(
             dict(x=X, y=y, sample_weight=sample_weight)
         )
-        dataset_meta = getattr(self.dataset_transformer_, "get_metadata", dict)()
-        vars(self).update(**dataset_meta)
 
         return X, y, sample_weight
 

Original file line number	Diff line number	Diff line change
`@@ -817,8 +817,6 @@ def _initialize(`
`817`	`817`	`self.dataset_transformer_ = self.dataset_transformer.fit(`
`818`	`818`	`dict(x=X, y=y, sample_weight=sample_weight)`
`819`	`819`	`)`
`820`		`- dataset_meta = getattr(self.dataset_transformer_, "get_metadata", dict)()`
`821`		`- vars(self).update(**dataset_meta)`
`822`	`820`
`823`	`821`	`return X, y, sample_weight`
`824`	`822`