fix: update test_batch to enforce classification in basic test

adaptive-machine-learning · May 24, 2024 · bda30f5 · bda30f5
1 parent d8c44d8
commit bda30f5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_batch.py b/tests/test_batch.py
@@ -67,7 +67,7 @@ def test_batch_basic():
     y = np.arange(n)
     assert x.shape == (n, feature_count)
 
-    stream = NumpyStream(x, y)
+    stream = NumpyStream(x, y, target_type='categorical')
     learner = _DummyBatchClassifierSSL(batch_size, stream.schema, class_value_type=str)
     prequential_ssl_evaluation(
         stream=stream, learner=learner, label_probability=0.01, window_size=100