Skip to content

Commit de7da65

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 8d6b623 commit de7da65

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

mesa_frames/abstract/agents.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1126,7 +1126,9 @@ def index(self) -> Index: ...
11261126

11271127
@property
11281128
def pos(self) -> DataFrame:
1129-
pos = self._df_get_masked_df(df=self.space.agents, index_cols="agent_id", mask=self.index)
1129+
pos = self._df_get_masked_df(
1130+
df=self.space.agents, index_cols="agent_id", mask=self.index
1131+
)
11301132
pos = self._df_reindex(
11311133
pos, self.index, new_index_cols="unique_id", original_index_cols="agent_id"
11321134
)

mesa_frames/concrete/mixin.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -344,7 +344,9 @@ def _df_groupby_cumcount(
344344
def _df_index(self, df: pl.DataFrame, index_col: str) -> pl.Series: ...
345345

346346
@overload
347-
def _df_index(self, df: pl.DataFrame, index_col: Collection[str]) -> pl.DataFrame: ...
347+
def _df_index(
348+
self, df: pl.DataFrame, index_col: Collection[str]
349+
) -> pl.DataFrame: ...
348350

349351
def _df_index(
350352
self, df: pl.DataFrame, index_col: str | Collection[str]

0 commit comments

Comments
 (0)