Fix: do not propagate kwargs to super().__init__ of KerasBaseEstimator

equinor · Jun 25, 2024 · 17ad386 · 17ad386
1 parent ba15f75
commit 17ad386
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/gordo/serializer/test_serializer_from_definition.py b/tests/gordo/serializer/test_serializer_from_definition.py
@@ -278,4 +278,4 @@ def _verify_pipe(self, pipe, model, model_kind):
         # STEP 4 TEST:  Finally, the last step should be a KerasModel
         step4 = pipe.steps[3][1]
         self.assertIsInstance(step4, model)
-        self.assertTrue(step4._kind, model_kind)
+        self.assertTrue(step4.kind, model_kind)