From 3d500a5da00a711bff0d3ed2f5f3039527f92110 Mon Sep 17 00:00:00 2001 From: Martin Lang Date: Mon, 20 Feb 2023 17:05:25 +0100 Subject: [PATCH] Filter _step_files based on table data --- micromagneticdata/oommfdrive.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/micromagneticdata/oommfdrive.py b/micromagneticdata/oommfdrive.py index 0381c5f..9e3aeec 100644 --- a/micromagneticdata/oommfdrive.py +++ b/micromagneticdata/oommfdrive.py @@ -86,7 +86,9 @@ def _table_path(self): @property def _step_file_glob(self): - return self.drive_path.glob(f"{self.name}*.omf") + step_files = self.drive_path.glob(f"{self.name}*.omf") + column = self.table.data["iteration"].to_numpy() + return filter(lambda step: int(step.stem.split("-")[-1]) in column, step_files) @property def calculator_script(self):