fix name of _drop_level_feats in test

reichlab · Nov 7, 2024 · 33d721e · 33d721e
1 parent e488f6a
commit 33d721e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit/gbqr/test_drop_level_feats.py b/tests/unit/gbqr/test_drop_level_feats.py
@@ -1,4 +1,4 @@
-from idmodels.preprocess import drop_level_feats
+from idmodels.preprocess import _drop_level_feats
 
 
 def test_drop_level_feats():
@@ -51,7 +51,7 @@ def test_drop_level_feats():
                 "inc_trans_cs_taylor_d1_c1_w5t_sNone_lag1", "inc_trans_cs_taylor_d1_c1_w5t_sNone_lag2",
                 "horizon"]
 
-    actual = drop_level_feats(in_feats)
+    actual = _drop_level_feats(in_feats)
 
     assert len(actual) == len(expected)
     assert not set(actual) - set(expected)