fix integreation test with SDMetrics-0.16.0

sdv-dev · Sep 27, 2024 · ab3bcc0 · ab3bcc0
1 parent ca381a7
commit ab3bcc0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration/evaluation/test_multi_table.py b/tests/integration/evaluation/test_multi_table.py
@@ -60,7 +60,7 @@ def test_evaluation():
 def test_evaluation_metadata():
     """Test ``evaluate_quality`` and ``run_diagnostic`` with Metadata."""
     # Setup
-    table = pd.DataFrame({'id': [0, 1, 2, 3], 'col': [1, 2, 3, 4]})
+    table = pd.DataFrame({'id': [0, 1, 2, 3], 'col': [1, 2, 3, 4.0]})
     slightly_different_table = pd.DataFrame({'id': [0, 1, 2, 3], 'col': [1, 2, 3, 3.5]})
     data = {
         'table1': table,