iter

Author: glemaitre

imblearn/keras/_generator.py

@@ -33,7 +33,10 @@ def import_from_tensforflow():
     ParentClassTensorflow, has_keras_tf = import_from_tensforflow()
     has_keras = has_keras_k or has_keras_tf
     if has_keras:
-        ParentClass = ParentClassKeras + ParentClassTensorflow
+        if has_keras_tf:
+            ParentClass = ParentClassTensorflow
+        else:
+            ParentClass = ParentClassKeras
     else:
         ParentClass = (object,)
     return ParentClass, has_keras