Reduce the _load_runtime_context call lock

[Chise1]

May be _load_runtime_context can be better:

def _load_runtime_context(func: _F) -> _F:
    @wraps(func)  # type: ignore[misc]
    def wrapper(  # type: ignore[misc]
        *args: typing.Tuple[typing.Any, typing.Any],
        **kwargs: typing.Dict[typing.Any, typing.Any],
    ) -> typing.Optional[typing.Any]:
        global _RUNTIME_CONTEXT  # pylint: disable=global-statement
        if _RUNTIME_CONTEXT is None:
            with _RUNTIME_CONTEXT_LOCK:
                if _RUNTIME_CONTEXT is not None:
                    return
               ...

[pmcollins]

Hi @Chise1 does this PR address your concerns?

[Chise1]

yes.

[pmcollins]

Thanks -- can we close the related PR as well?