Fix DayZSynthesizer.create_parameters errors in Colab with numeric columns (#2706)

Author: frances-h

sdv/single_table/_dayz_utils.py

@@ -42,8 +42,8 @@ def detect_column_parameters(data, metadata, table_name):
         if sdtype == 'numerical':
             column_parameters[column_name] = {
                 'num_decimal_digits': learn_rounding_digits(data[column_name]),
-                'min_value': data[column_name].min().item(),
-                'max_value': data[column_name].max().item(),
+                'min_value': data[column_name].min(),
+                'max_value': data[column_name].max(),
             }
         elif sdtype == 'datetime':
             datetime_format = column_metadata.get('datetime_format', None)
@@ -63,13 +63,13 @@ def detect_column_parameters(data, metadata, table_name):
                 'start_timestamp': start_timestamp,
                 'end_timestamp': end_timestamp,
             }
-        elif sdtype in ['categorical', 'boolean']:
+        elif sdtype == 'categorical':
             column_parameters[column_name] = {
                 'category_values': data[column_name].dropna().unique().tolist()
             }
 
-        column_parameters[column_name]['missing_values_proportion'] = (
-            data[column_name].isna().mean().item()
+        column_parameters[column_name]['missing_values_proportion'] = float(
+            data[column_name].isna().mean()
         )
 
     return {'columns': column_parameters}

tests/integration/multi_table/test_dayz.py

@@ -24,7 +24,6 @@ def test_create_parameters_end_to_end(self):
                         'guest_email': {'missing_values_proportion': 0.0},
                         'hotel_id': {'missing_values_proportion': 0.0},
                         'has_rewards': {
-                            'category_values': [False, True],
                             'missing_values_proportion': 0.0,
                         },
                         'room_type': {

tests/integration/single_table/test_dayz.py

@@ -23,7 +23,6 @@ def test_create_parameters_end_to_end(self):
                     'columns': {
                         'guest_email': {'missing_values_proportion': 0.0},
                         'has_rewards': {
-                            'category_values': [False, True],
                             'missing_values_proportion': 0.0,
                         },
                         'room_type': {