fix (#842)

pymc-labs · Sep 10, 2024 · b75bd10 · b75bd10
1 parent ba5fa02
commit b75bd10
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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