ruff: enable ICN and PD rules

This commit is contained in:
Dima Gerasimov 2024-08-28 03:09:32 +01:00 committed by karlicoss
parent c5df3ce128
commit fc0e0be291
3 changed files with 7 additions and 3 deletions

View file

@ -83,7 +83,7 @@ def make_dataframe(data: Iterator[Result]):
}
df = pd.DataFrame(it())
df.set_index('dt', inplace=True)
df = df.set_index('dt')
# TODO not sure about UTC??
df.index = pd.to_datetime(df.index, utc=True)
return df

View file

@ -155,9 +155,9 @@ def dataframe() -> DataFrameT:
last = s # meh
dicts.append(d)
import pandas
import pandas as pd
return pandas.DataFrame(dicts)
return pd.DataFrame(dicts)
def stats() -> Stats:

View file

@ -9,9 +9,11 @@ lint.extend-select = [
"C4", # flake8-comprehensions -- unnecessary list/map/dict calls
"COM", # trailing commas
"EXE", # various checks wrt executable files
"ICN", # various import conventions
"FBT", # detect use of boolean arguments
"FURB", # various rules
"PERF", # various potential performance speedups
"PD", # pandas rules
"PIE", # 'misc' lints
"PLR", # 'refactor' rules
"PLW", # pylint warnings
@ -104,4 +106,6 @@ lint.ignore = [
"PT012", # pytest raises should contain a single statement
"COM812", # trailing comma missing -- TODO maybe use this?
"PD901", # generic variable name df
]