|
16 | 16 | import uuid |
17 | 17 | from dataclasses import dataclass |
18 | 18 | from inspect import signature |
| 19 | +from typing import Any |
19 | 20 |
|
20 | 21 | import numpy as np |
21 | 22 | from geoh5py.data import Data |
@@ -140,9 +141,7 @@ def write_files(self, lookup): |
140 | 141 |
|
141 | 142 | ifile = InputFile.read_ui_json(self.params.worker_uijson) |
142 | 143 | with ifile.data["geoh5"].open(mode="r") as workspace: |
143 | | - |
144 | 144 | for name, trial in lookup.items(): |
145 | | - |
146 | 145 | if trial["status"] != "pending": |
147 | 146 | continue |
148 | 147 |
|
@@ -193,11 +192,14 @@ def call_worker(ifile: InputFile): |
193 | 192 |
|
194 | 193 | run_cmd = ifile.data["run_command"] |
195 | 194 | module = importlib.import_module(run_cmd) |
196 | | - filt = ( |
197 | | - lambda member: inspect.isclass(member) |
198 | | - and member.__module__ == run_cmd |
199 | | - and hasattr(member, "run") |
200 | | - ) |
| 195 | + |
| 196 | + def filt(member: Any) -> bool: |
| 197 | + return ( |
| 198 | + inspect.isclass(member) |
| 199 | + and member.__module__ == run_cmd |
| 200 | + and hasattr(member, "run") |
| 201 | + ) |
| 202 | + |
201 | 203 | driver = inspect.getmembers(module, filt)[0][1] |
202 | 204 | driver.start(ifile.path_name) |
203 | 205 |
|
@@ -226,7 +228,6 @@ def main(file_path): |
226 | 228 |
|
227 | 229 |
|
228 | 230 | if __name__ == "__main__": |
229 | | - |
230 | 231 | parser = argparse.ArgumentParser( |
231 | 232 | description="Run parameter sweep of worker driver." |
232 | 233 | ) |
|
0 commit comments