From 66f675024abb200479535a0fa43f410013f01d2a Mon Sep 17 00:00:00 2001 From: Juan Orduz Date: Fri, 19 Jul 2024 18:57:00 +0200 Subject: [PATCH] fix (#842) --- .pre-commit-config.yaml | 2 +- tests/test_model_builder.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index b9f274319..05af4713c 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -11,7 +11,7 @@ repos: - --exclude=docs/ - --exclude=scripts/ - repo: https://github.com/astral-sh/ruff-pre-commit - rev: v0.5.1 + rev: v0.5.3 hooks: - id: ruff types_or: [ python, pyi, jupyter ] diff --git a/tests/test_model_builder.py b/tests/test_model_builder.py index 0403263bf..28cd4c80c 100644 --- a/tests/test_model_builder.py +++ b/tests/test_model_builder.py @@ -258,7 +258,7 @@ def test_predict(fitted_model_instance): prediction_data = pd.DataFrame({"input": x_pred}) pred = fitted_model_instance.predict(prediction_data) # Perform elementwise comparison using numpy - assert type(pred) == np.ndarray + assert isinstance(pred, np.ndarray) assert len(pred) > 0