Update test to reflect the new treatment col (#133)

* Don't infer ncust, allow for easy override * Add treatment support, even better rounding * Update test to reflect the new treatment col
pegasystems · Oct 4, 2023 · 8a6a81e · 8a6a81e
1 parent b66f402
commit 8a6a81e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tests/test_datasets.py b/python/tests/test_datasets.py
@@ -31,4 +31,4 @@ def test_import_SampleTrees():
 
 def test_import_SampleValueFinder():
     vf = datasets.SampleValueFinder()
-    assert vf.df.shape == (27133, 10)
+    assert vf.df.shape == (27133, 11)