fix

aimclub · Feb 14, 2024 · a0647af · a0647af
1 parent ba97629
commit a0647af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit_test.py b/tests/unit_test.py
@@ -20,7 +20,7 @@ def test_dataset_info(self):
         self.assertIsInstance(result[1], float)
         self.assertIsInstance(result[2], float)
         self.assertIsInstance(result[3], float)
-        self.assertIsInstance(result[4], float)
+        self.assertIsInstance(result[4], int)
 
 if __name__ == "__main__":
     unittest.main()