Fix pytorch_state_dict import from library

[jdeschamps]

Issue

PyTorchModelAdaptor passes a sha256 keyword to the callable responsible for instantiating the model:

core-bioimage-io-python/bioimageio/core/model_adapters/_pytorch_model_adapter.py

Line 107 in ff8af21

sha256=(

If the callable is an ArchitectureFromLibraryDescr, then an error will be raised since there is no kwargs in the function signature:

core-bioimage-io-python/bioimageio/core/digest_spec.py

Line 64 in ff8af21

def _(node: ArchitectureFromLibraryDescr) -> Callable[..., Any]:

Fix

Add kwargs to the function signature (and that of other callables).

core-bioimage-io-python/bioimageio/core/digest_spec.py

Line 65 in 7681ab7

node: ArchitectureFromLibraryDescr, **kwargs: Unpack[HashKwargs]

SIde note

The pre-commit hooks forced me to fix an import, is it due to me not having the right version of the specs?

core-bioimage-io-python/bioimageio/core/digest_spec.py

Line 23 in 7681ab7

from bioimageio.spec._internal.io import HashKwargs, download

[FynnBe]

The pre-commit hooks forced me to fix an import, is it due to me not having the right version of the specs?

no, not sure why this wasn't picked up before.. I suppose it was a ruff check? the fix is good anyway.

[FynnBe]

nice fix!
sad that singledispatch does play nice with static type checking... we should probably remove it in favor of some if else switch to have this part statically checked.