'int' in integration test

sdv-dev · Sep 7, 2023 · afbf35d · afbf35d
1 parent 054d9eb
commit afbf35d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration/transformers/test_base.py b/tests/integration/transformers/test_base.py
@@ -155,7 +155,7 @@ def _reverse_transform(self, data):
             result = data.diff(axis=1)
             result.iloc[:, 0] = data.iloc[:, 0]
 
-            return result.astype(int)
+            return result.astype('int64')
 
     data_test = pd.DataFrame({
         'col_1': [1, 2, 3],