Skip to content

Commit

Permalink
👷 mypy
Browse files Browse the repository at this point in the history
  • Loading branch information
juftin committed Dec 15, 2023
1 parent f91e784 commit ed8f778
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion lunchable/plugins/base/pandas_app.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ def df_to_models(
array_df = df.copy()
array_df = array_df.fillna(np.NaN).replace([np.NaN], [None])
model_array = array_df.to_dict(orient="records")
return [model_type(**item) for item in model_array] # type: ignore[misc]
return [model_type(**item) for item in model_array]


class LunchablePandasTransactionsApp(LunchableTransactionsApp, LunchablePandasApp):
Expand Down
4 changes: 2 additions & 2 deletions lunchable/plugins/primelunch/primelunch.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ def amazon_to_df(self) -> pd.DataFrame:
-------
pd.DataFrame
"""
dt64: np.dtype[datetime64] = np.dtype("datetime64[ns]") # type: ignore[assignment]
dt64: np.dtype[datetime64] = np.dtype("datetime64[ns]")
expected_columns = {
"order id": str,
"items": str,
Expand All @@ -66,7 +66,7 @@ def amazon_to_df(self) -> pd.DataFrame:
"refund": np.float64,
"payments": str,
}
amazon_df = pd.read_csv( # type: ignore[call-overload]
amazon_df = pd.read_csv(
self.file_path,
usecols=expected_columns.keys(),
)
Expand Down

0 comments on commit ed8f778

Please sign in to comment.