Compare ufunc with CPU only for now

Signed-off-by: Julio Faracco <[email protected]>
discovery-unicamp · Feb 12, 2024 · efc902a · efc902a
1 parent c42c4b1
commit efc902a
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/datasets/test_generic.py b/tests/datasets/test_generic.py
@@ -67,8 +67,8 @@ def test_add(self):
         dataset1 = DatasetArray(name="Array", root=raw_path, download=False)
         dataset2 = DatasetArray(name="Array", root=raw_path, download=False)
 
-        dataset1.load()
-        dataset2.load()
+        dataset1._load_cpu()
+        dataset2._load_cpu()
 
         np1 = np.load(raw_path)
         np2 = np.load(raw_path)
@@ -87,8 +87,8 @@ def test_sub(self):
         dataset1 = DatasetArray(name="Array", root=raw_path, download=False)
         dataset2 = DatasetArray(name="Array", root=raw_path, download=False)
 
-        dataset1.load()
-        dataset2.load()
+        dataset1._load_cpu()
+        dataset2._load_cpu()
 
         np1 = np.load(raw_path)
         np2 = np.load(raw_path)
@@ -107,8 +107,8 @@ def test_mul(self):
         dataset1 = DatasetArray(name="Array", root=raw_path, download=False)
         dataset2 = DatasetArray(name="Array", root=raw_path, download=False)
 
-        dataset1.load()
-        dataset2.load()
+        dataset1._load_cpu()
+        dataset2._load_cpu()
 
         np1 = np.load(raw_path)
         np2 = np.load(raw_path)