From f3637882f9ed5d29037179de47d72f9e42a17da7 Mon Sep 17 00:00:00 2001
From: Spencer Sun <spencer@Spencers-MacBook-Pro.local>
Date: Wed, 22 May 2024 19:48:33 +1200
Subject: [PATCH] fix: fix stream consistency

---
 src/capymoa/stream/_stream.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/capymoa/stream/_stream.py b/src/capymoa/stream/_stream.py
index a58c1e07..a9e5bccb 100644
--- a/src/capymoa/stream/_stream.py
+++ b/src/capymoa/stream/_stream.py
@@ -27,7 +27,7 @@ def _target_is_categorical(targets, target_type):
     if target_type is None:
         if type(targets[0]) == str or type(targets[0]) == bool:
             return True
-        if type(targets[0]) == np.float64:
+        if type(targets[0]) == np.float64 or type(targets[0]) == np.int64:
             num_unique = len(np.unique(targets))
             if num_unique >= 20:
                 warnings.warn(f'target variable includes {num_unique} (≥ 20) unique values, inferred as numeric, '