Skip to content

Commit

Permalink
linting fix
Browse files Browse the repository at this point in the history
  • Loading branch information
trevorb1 committed Oct 21, 2023
1 parent 019a096 commit 2890770
Showing 1 changed file with 11 additions and 3 deletions.
14 changes: 11 additions & 3 deletions src/otoole/input.py
Original file line number Diff line number Diff line change
Expand Up @@ -138,12 +138,20 @@ def _add_dtypes(self, config: Dict):
dtypes = {}
for column in details["indices"] + ["VALUE"]:
if column == "VALUE":
dtypes["VALUE"] = details["dtype"] if details["dtype"] != "int" else "int64"
dtypes["VALUE"] = (
details["dtype"] if details["dtype"] != "int" else "int64"
)
else:
dtypes[column] = config[column]["dtype"] if config[column]["dtype"] != "int" else "int64"
dtypes[column] = (
config[column]["dtype"]
if config[column]["dtype"] != "int"
else "int64"
)
details["index_dtypes"] = dtypes
elif details["type"] == "set":
details["dtype"] = details["dtype"] if details["dtype"] != "int" else "int64"
details["dtype"] = (
details["dtype"] if details["dtype"] != "int" else "int64"
)
return config

@property
Expand Down

0 comments on commit 2890770

Please sign in to comment.