Fix integration test

sdv-dev · Aug 7, 2024 · 2e25c3f · 2e25c3f
1 parent 6ea14cf
commit 2e25c3f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/integration/evaluation/test_single_table.py b/tests/integration/evaluation/test_single_table.py
@@ -36,8 +36,8 @@ def test_evaluation_metadata():
     """Test ``evaluate_quality`` and ``run_diagnostic`` with Metadata."""
     # Setup
     data = pd.DataFrame({'col': [1, 2, 3]})
-    metadata = Metadata()
-    metadata.add_column('col', sdtype='numerical')
+    metadata_dict = {'columns': {'col': {'sdtype': 'numerical'}}}
+    metadata = Metadata.load_from_dict(metadata_dict)
     synthesizer = GaussianCopulaSynthesizer(metadata, default_distribution='truncnorm')
 
     # Run and Assert