Fix test

sdv-dev · Mar 7, 2025 · 822bcb9 · 822bcb9
1 parent 2cae294
commit 822bcb9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit/cag/test_range.py b/tests/unit/cag/test_range.py
@@ -763,7 +763,7 @@ def test_reverse_transform_floats(self):
                 'a': np.dtype('float'),
                 'b': np.dtype('float'),
                 'c': np.dtype('float'),
-                'col': np.dtype('int'),
+                'col': pd.Series([1]).dtype,
             }
         }
 
@@ -805,7 +805,7 @@ def test_reverse_transform_datetime(self):
                 'a': np.dtype('<M8[ns]'),
                 'b': np.dtype('<M8[ns]'),
                 'c': np.dtype('<M8[ns]'),
-                'col': np.dtype('int64'),
+                'col': pd.Series([1]).dtype,
             }
         }