Put the correct test value

nubank · Aug 15, 2023 · cc9031e · cc9031e
1 parent c8eacd5
commit cc9031e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/metrics/test_pd_extractors.py b/tests/metrics/test_pd_extractors.py
@@ -216,7 +216,7 @@ def test_extract():
     assert extract(tlc_results, base_extractors).shape == (12, 9)
     assert extract(tlc_results, splitter_extractor).shape == (36, 10)
 
-    assert extract(sc_results, base_extractors).shape == (5, 9)
+    assert extract(sc_results, base_extractors).shape == (667, 9)
     assert extract(sc_results, splitter_extractor).shape == (15, 10)
 
     assert extract(fw_sc_results, base_extractors).shape == (3, 9)