From 33d721e09a6a7a856bb273cb2b8074c89c8b50b2 Mon Sep 17 00:00:00 2001 From: Evan Ray Date: Thu, 7 Nov 2024 11:37:11 -0500 Subject: [PATCH] fix name of _drop_level_feats in test --- tests/unit/gbqr/test_drop_level_feats.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/gbqr/test_drop_level_feats.py b/tests/unit/gbqr/test_drop_level_feats.py index 337fffe..6ddbfa6 100644 --- 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)