fix import

pyro-ppl · Oct 30, 2023 · b475541 · b475541
1 parent 7b958f1
commit b475541
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pyro/poutine/runtime.py b/pyro/poutine/runtime.py
@@ -7,15 +7,14 @@
 from typing import TYPE_CHECKING, Callable, Dict, List, Optional, Set, Tuple, Union
 
 import torch
+from typing_extensions import TypedDict
 
 from pyro.params.param_store import (  # noqa: F401
     _MODULE_NAMESPACE_DIVIDER,
     ParamStoreDict,
 )
 
 if TYPE_CHECKING:
-    from typing_extensions import TypedDict
-
     from pyro.poutine.indep_messenger import CondIndepStackFrame
     from pyro.poutine.messenger import Messenger