diff --git a/src/hydra_zen/wrapper/_implementations.py b/src/hydra_zen/wrapper/_implementations.py index c74c0f00..f9c38967 100644 --- a/src/hydra_zen/wrapper/_implementations.py +++ b/src/hydra_zen/wrapper/_implementations.py @@ -967,18 +967,18 @@ def get_name(target: _HasName) -> str: return name -class _StoreCallSig(TypedDict): - """Arguments for ZenStore.__call__ - - This default dict enables us to easily update/merge the default arguments for a - specific ZenStore instance, in support of self-partialing behavior.""" - - name: Union[NodeName, Callable[[Any], NodeName]] - group: Union[GroupName, Callable[[Any], GroupName]] - package: Optional[Union[str, Callable[[Any], str]]] - provider: Optional[str] - __kw: dict[str, Any] # kwargs passed to to_config - to_config: Callable[[Any], Any] +# Arguments for ZenStore.__call__ +# +# This default dict enables us to easily update/merge the default arguments for a +# specific ZenStore instance, in support of self-partialing behavior. +_StoreCallSig = TypedDict("_StoreCallSig", [ + ("name", Union[NodeName, Callable[[Any], NodeName]]), + ("group", Union[GroupName, Callable[[Any], GroupName]]), + ("package", Optional[Union[str, Callable[[Any], str]]]), + ("provider", Optional[str]), + ("__kw", dict[str, Any]), # kwargs passed to to_config + ("to_config", Callable[[Any], Any]), +]) # TODO: make frozen dict