From 822bcb959e51403f18608718ff37b90ab12b7e1d Mon Sep 17 00:00:00 2001
From: Felipe <fealho@gmail.com>
Date: Fri, 7 Mar 2025 10:15:23 -0800
Subject: [PATCH] Fix test

---
 tests/unit/cag/test_range.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/unit/cag/test_range.py b/tests/unit/cag/test_range.py
index f1ffcf551..c21f71709 100644
--- 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,
             }
         }