Skip to content

Commit ef03c8c

Browse files
committed
fix lint
1 parent b011d98 commit ef03c8c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

autointent/modules/scoring/_catboost/catboost_scorer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,8 @@ def fit(
176176
if self.features_type == FeaturesType.BOTH and self.use_embedding_features:
177177
extra_params["embedding_features"] = ["embedding"]
178178
else:
179-
raise ValueError(f"Unsupported features type: {self.features_type}")
179+
msg = f"Unsupported features type: {self.features_type}"
180+
raise ValueError(msg)
180181
self.catboost_kwargs.update(extra_params)
181182

182183
self._model = CatBoostClassifier(

tests/modules/scoring/test_catboost.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,6 @@ def test_catboost_features_types(dataset, features_type, use_embedding_features)
119119
]
120120

121121
predictions = scorer.predict(test_data)
122-
print(predictions)
123122
assert predictions.shape[0] == len(test_data)
124123
assert predictions.shape[1] == len(set(data_handler.train_labels(0)))
125124
assert 0.0 <= np.min(predictions) <= np.max(predictions) <= 1.0

0 commit comments

Comments
 (0)