PARSynthesizer is not learning rounding scheme for numerical columns (…

…#2289)
sdv-dev · Nov 13, 2024 · 20c1f28 · 20c1f28
1 parent 4d56fe7
commit 20c1f28
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/sdv/sequential/par.py b/sdv/sequential/par.py
@@ -87,7 +87,7 @@ def __init__(
         self,
         metadata,
         enforce_min_max_values=True,
-        enforce_rounding=False,
+        enforce_rounding=True,
         locales=['en_US'],
         context_columns=None,
         segment_size=None,

diff --git a/tests/integration/sequential/test_par.py b/tests/integration/sequential/test_par.py
@@ -59,6 +59,7 @@ def test_par():
     assert len(loss_values) == 1
     assert all(sampled.groupby('store_id')['date'].is_monotonic_increasing)
     assert all(sampled.groupby('store_id')['date'].agg(lambda x: x.is_unique))
+    assert all(sampled['total_sales'].round(2) == sampled['total_sales'])
 
 
 def test_column_after_date_simple():