From 426ba2ff234659843fcdfbd252858bb24dcfba97 Mon Sep 17 00:00:00 2001
From: mooniean <bmbcgomes@gmail.com>
Date: Thu, 27 Jun 2024 15:35:39 +0100
Subject: [PATCH] updated in1d to isin

---
 src/caked/dataloader.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/caked/dataloader.py b/src/caked/dataloader.py
index ee68c95..31dea69 100644
--- a/src/caked/dataloader.py
+++ b/src/caked/dataloader.py
@@ -100,13 +100,13 @@ def load(self, datapath, datatype) -> None:
         if len(self.classes) == 0:
             self.classes = ids
         else:
-            class_check = np.in1d(self.classes, ids)
+            class_check = np.isin(self.classes, ids)
             if not np.all(class_check):
                 msg = "Not all classes in the list are present in the directory. Missing classes: {}".format(
                     np.asarray(self.classes)[~class_check]
                 )
                 raise RuntimeError(msg)
-            class_check = np.in1d(ids, self.classes)
+            class_check = np.isin(ids, self.classes)
             if not np.all(class_check):
                 logging.basicConfig(format="%(message)s", level=logging.INFO)
                 logging.info(