Skip to content

Commit

Permalink
fix pyright
Browse files Browse the repository at this point in the history
Signed-off-by: zjgemi <liuxin_zijian@163.com>
  • Loading branch information
zjgemi committed Oct 22, 2024
1 parent 3b453f4 commit a20cea0
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions dpgen2/exploration/selector/conf_filter.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,9 +72,10 @@ def check(
self,
ms: dpdata.MultiSystems,
) -> dpdata.MultiSystems:
selected_idx = sum(
[[(i, j) for j in range(s.get_nframes())] for i, s in enumerate(ms)], []
)
selected_idx = []
for i in range(len(ms)):
for j in range(ms[i].get_nframes()):
selected_idx.append((i, j))
for ff in self._filters:
res = ff.batched_check([ms[i][j] for i, j in selected_idx])
selected_idx = [idx for i, idx in enumerate(selected_idx) if res[i]]
Expand Down

0 comments on commit a20cea0

Please sign in to comment.