Skip to content

Commit 6f0c281

Browse files
authored
Allow source input request overrides to be list (#97)
1 parent f4b59a8 commit 6f0c281

File tree

2 files changed

+18
-6
lines changed

2 files changed

+18
-6
lines changed

src/pproc/config/base.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -352,7 +352,11 @@ def sort_inputs(cls, inputs: list[dict]) -> dict:
352352

353353
@classmethod
354354
def _input_request(
355-
cls, src_name: str, requests: list[dict], accum_dims: list[str], **overrides
355+
cls,
356+
src_name: str,
357+
requests: list[dict],
358+
accum_dims: list[str],
359+
overrides: list | dict,
356360
) -> dict | list[dict]:
357361
[req.pop(dim, None) for req in requests for dim in accum_dims]
358362
updated_inputs = [
@@ -372,7 +376,7 @@ def _populate_inputs(
372376
src_overrides = overrides.get(src_name, {}).copy()
373377
ret[src_name] = {
374378
"request": cls._input_request(
375-
src_name, requests, accum_dims, **src_overrides.pop("request", {})
379+
src_name, requests, accum_dims, src_overrides.pop("request", {})
376380
),
377381
**src_overrides,
378382
}

src/pproc/config/types.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -560,12 +560,16 @@ def from_schema(cls, schema_config: dict, **overrides) -> Self:
560560

561561
@classmethod
562562
def _input_request(
563-
cls, src_name: str, requests: list[dict], accum_dims: list[str], **overrides
563+
cls,
564+
src_name: str,
565+
requests: list[dict],
566+
accum_dims: list[str],
567+
overrides: list | dict,
564568
) -> dict | list[dict]:
565569
if src_name == "clim":
566570
accum_dims = accum_dims.copy()
567571
accum_dims.remove("step")
568-
return super()._input_request(src_name, requests, accum_dims, **overrides)
572+
return super()._input_request(src_name, requests, accum_dims, overrides)
569573

570574
@classmethod
571575
def sort_inputs(cls, inputs: list[dict]) -> dict:
@@ -701,12 +705,16 @@ def _format_out(self, param: ParamConfig, req: dict) -> dict:
701705

702706
@classmethod
703707
def _input_request(
704-
cls, src_name: str, requests: list[dict], accum_dims: list[str], **overrides
708+
cls,
709+
src_name: str,
710+
requests: list[dict],
711+
accum_dims: list[str],
712+
overrides: list | dict,
705713
) -> dict | list[dict]:
706714
if src_name == "clim":
707715
accum_dims = accum_dims.copy()
708716
accum_dims.remove("step")
709-
return super()._input_request(src_name, requests, accum_dims, **overrides)
717+
return super()._input_request(src_name, requests, accum_dims, overrides)
710718

711719
@classmethod
712720
def sort_inputs(cls, inputs: list[dict]) -> dict:

0 commit comments

Comments
 (0)