Propapagate type aliases through sphinx (#453)

* type annotations fix in dynamical systems

* add type aliases for interventional

* add type aliases for observational

* add counterfactual doc type aliases

* lint

Author: SamWitty

chirho/counterfactual/handlers/counterfactual.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Any, Dict, Generic, Mapping, TypeVar
 
 import pyro

chirho/counterfactual/handlers/explanation.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import collections.abc
 import contextlib
 import functools

chirho/counterfactual/ops.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import functools
 from typing import Optional, Tuple, TypeVar
 

chirho/dynamical/handlers/interruption.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import numbers
 import typing
 from typing import Callable, Generic, Tuple, TypeVar, Union

chirho/dynamical/ops.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import numbers
 from typing import Callable, Mapping, Optional, Tuple, TypeVar, Union
 

chirho/interventional/handlers.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import collections
 import functools
 from typing import Callable, Dict, Generic, Hashable, Mapping, Optional, TypeVar
@@ -41,7 +43,7 @@ def _intervene_atom(
 def _intervene_atom_distribution(
     obs: pyro.distributions.Distribution,
     act: Optional[AtomicIntervention[pyro.distributions.Distribution]] = None,
-    **kwargs
+    **kwargs,
 ) -> pyro.distributions.Distribution:
     """
     Intervene on a distribution in a probabilistic program.
@@ -70,7 +72,7 @@ def _dict_intervene(
 def _intervene_callable(
     obs: collections.abc.Callable,
     act: Optional[CompoundIntervention[T]] = None,
-    **call_kwargs
+    **call_kwargs,
 ) -> Callable[..., T]:
     if act is None:
         return obs

chirho/interventional/ops.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import functools
 from typing import Callable, Hashable, Mapping, Optional, Tuple, TypeVar, Union
 

chirho/observational/handlers/condition.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Callable, Generic, Hashable, Mapping, TypeVar, Union
 
 import pyro

chirho/observational/handlers/soft_conditioning.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import functools
 import operator
 from typing import Callable, Literal, Optional, Protocol, TypedDict, TypeVar, Union

chirho/observational/internals.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from typing import Mapping, Optional, TypeVar
 
 import pyro