Support SDMetrics-0.16.0 (#2236)

sdv-dev · Sep 25, 2024 · 2ec3eaa · 2ec3eaa
1 parent 2f9c842
commit 2ec3eaa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -36,7 +36,7 @@ dependencies = [
     'ctgan>=0.10.0',
     'deepecho>=0.6.0',
     'rdt>=1.12.3',
-    'sdmetrics>=0.14.0',
+    'sdmetrics>=0.16.0',
     'platformdirs>=4.0',
     'pyyaml>=6.0.1',
 ]

diff --git a/tests/integration/evaluation/test_multi_table.py b/tests/integration/evaluation/test_multi_table.py
@@ -7,7 +7,7 @@
 def test_evaluation():
     """Test ``evaluate_quality`` and ``run_diagnostic``."""
     # 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,