Fix NotImplementedError when using FloatFormatter with numerical data types during fit due to PyArrow

pyproject.toml

@@ -58,6 +58,7 @@ test = [
     'rundoc>=0.4.3,<0.5',
     'pytest-subtests>=0.5,<1.0',
     'pytest-runner >= 2.11.1',
+    'pyarrow >= 17.0.0',
     'tomli>=2.0.0,<3',
 ]
 dev = [

rdt/transformers/numerical.py

@@ -104,7 +104,7 @@ def _raise_out_of_bounds_error(self, value, name, bound_type, min_bound, max_bou
 
     def _validate_values_within_bounds(self, data):
         if not self.computer_representation.startswith('Float'):
-            fractions = data[~data.isna() & data % 1 != 0]
+            fractions = data[~data.isna() & (data != (data // 1))]
             if not fractions.empty:
                 raise ValueError(
                     f"The column '{data.name}' contains float values {fractions.tolist()}. "

rdt/transformers/utils.py

@@ -261,7 +261,7 @@ def learn_rounding_digits(data):
         return None
 
     # Doesn't contain decimal digits
-    if ((roundable_data % 1) == 0).all():
+    if (roundable_data == roundable_data.astype(int)).all():
         return 0
 
     # Try to round to fewer digits