diff --git a/newa/__init__.py b/newa/__init__.py index e029ec1..02e6462 100644 --- a/newa/__init__.py +++ b/newa/__init__.py @@ -342,10 +342,10 @@ class Recipe(Cloneable, Serializable): class RecipeConfig(Cloneable, Serializable): """ A job recipe configuration """ - fixtures: dict[str, dict] = field(factory=dict) - dimensions: dict[str, dict] = field(factory=dict) + fixtures: dict[str, dict[str, Any]] = field(factory=dict) + dimensions: dict[str, dict[str, Any]] = field(factory=dict) - def build_requests(self) -> list: + def build_requests(self) -> list[Request]: # this is here to generate unique recipe IDs recipe_id_gen = itertools.count(start=1) @@ -360,7 +360,7 @@ def build_requests(self) -> list: for i in range(len(combinations)): combinations[i] = (self.fixtures,) + (combinations[i]) - def merge_combination_data(combination): + def merge_combination_data(combination: list[dict[str, Any]]) -> dict[str, Any]: merged = {} for record in combination: for key in record: