diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 7ffa0612..95ac35ea 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -25,7 +25,7 @@ repos: exclude: &exclude_pattern 'iv_weak_instruments.ipynb' args: ["--maxkb=1500"] - repo: https://github.com/astral-sh/ruff-pre-commit - rev: v0.5.4 + rev: v0.5.5 hooks: # Run the linter - id: ruff diff --git a/causalpy/tests/test_synthetic_data.py b/causalpy/tests/test_synthetic_data.py index 0eb66f2e..51da6a3e 100644 --- a/causalpy/tests/test_synthetic_data.py +++ b/causalpy/tests/test_synthetic_data.py @@ -26,7 +26,7 @@ def test_generate_multicell_geolift_data(): from causalpy.data.simulate_data import generate_multicell_geolift_data df = generate_multicell_geolift_data() - assert type(df) == pd.DataFrame + assert isinstance(df, pd.DataFrame) assert np.all(df >= 0), "Found negative values in dataset" @@ -37,5 +37,5 @@ def test_generate_geolift_data(): from causalpy.data.simulate_data import generate_geolift_data df = generate_geolift_data() - assert type(df) == pd.DataFrame + assert isinstance(df, pd.DataFrame) assert np.all(df >= 0), "Found negative values in dataset"