diff --git a/tests/python_package_test/test_basic.py b/tests/python_package_test/test_basic.py index a0ce5b8f8b66..b499ca7b1ea2 100644 --- a/tests/python_package_test/test_basic.py +++ b/tests/python_package_test/test_basic.py @@ -215,7 +215,6 @@ def test_add_features_from_different_sources(self): # test that method works for different data types d1 = lgb.Dataset(x_1, feature_name=names, free_raw_data=False).construct() res_feature_names = [name for name in names] - idx = 1 for idx, x_2 in enumerate(xxs, 2): original_type = type(d1.get_data()) d2 = lgb.Dataset(x_2, feature_name=names, free_raw_data=False).construct() @@ -223,7 +222,6 @@ def test_add_features_from_different_sources(self): self.assertIsInstance(d1.get_data(), original_type) self.assertTupleEqual(d1.get_data().shape, (n_row, n_col * idx)) res_feature_names += ['D{}_{}'.format(idx, name) for name in names] - idx += 1 self.assertListEqual(d1.feature_name, res_feature_names) def test_cegb_affects_behavior(self):