chore: enforce more ruff rules (#31447)

Co-authored-by: Elizabeth Thompson <eschutho@gmail.com>
This commit is contained in:
Maxime Beauchemin
2024-12-18 17:41:34 -08:00
committed by GitHub
parent 9da65d6bfd
commit e51b95ffa8
375 changed files with 1821 additions and 1718 deletions

View File

@@ -28,7 +28,7 @@ from tests.unit_tests.fixtures.dataframes import prophet_df
def test_prophet_valid():
df = prophet(df=prophet_df, time_grain="P1M", periods=3, confidence_interval=0.9)
columns = {column for column in df.columns}
columns = {column for column in df.columns} # noqa: C416
assert columns == {
DTTM_ALIAS,
"a__yhat",
@@ -112,7 +112,7 @@ def test_prophet_valid():
def test_prophet_valid_zero_periods():
df = prophet(df=prophet_df, time_grain="P1M", periods=0, confidence_interval=0.9)
columns = {column for column in df.columns}
columns = {column for column in df.columns} # noqa: C416
assert columns == {
DTTM_ALIAS,
"a__yhat",