iter

Author: glemaitre

imblearn/keras/tests/test_generator.py

@@ -24,7 +24,8 @@ def data():
     X, y = make_imbalance(
         iris.data, iris.target, sampling_strategy={0: 30, 1: 50, 2: 40}
     )
-    y = LabelBinarizer().fit_transform(y)
+    X = X.astype(np.float32)
+    y = LabelBinarizer().fit_transform(y).astype(np.int32)
     return X, y
 
 
@@ -103,7 +104,7 @@ def test_balanced_batch_generator_function_no_return_indices(data):
         (None, None),
         (RandomOverSampler(), None),
         (NearMiss(), None),
-        (None, np.random.uniform(size=120)),
+        (None, np.random.uniform(size=120).astype(np.float32)),
     ],
 )
 def test_balanced_batch_generator_function(data, sampler, sample_weight):
@@ -117,6 +118,7 @@ def test_balanced_batch_generator_function(data, sampler, sample_weight):
         batch_size=10,
         random_state=42,
     )
+    print(next(training_generator))
     model.fit(
         training_generator,
         steps_per_epoch=steps_per_epoch,

pixi.lock

@@ -137,7 +137,7 @@ platforms = ["linux-64", "osx-arm64", "osx-64"]
 
 [tool.pixi.feature.tensorflow.dependencies]
 tensorflow = ">=2.13.1,<2.18"
-keras = ">=3.0.5,<3.8"
+keras = ">=3.0.5,<3.9"
 
 [tool.pixi.feature.min-dependencies.dependencies]
 numpy = "==1.24.3"