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

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

chirho/observational/ops.py

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

docs/source/conf.py

@@ -23,6 +23,22 @@
 author = 'Basis'
 
 
+# -- Type hints configuration ------------------------------------------------
+
+autodoc_type_aliases = {
+    'R': 'R',
+    'State': 'State',
+    'Dynamics': 'Dynamics',
+    'AtomicIntervention': 'AtomicIntervention',
+    'CompoundIntervention': 'CompoundIntervention',
+    'Intervention': 'Intervention',
+    'AtomicObservation': 'AtomicObservation',
+    'CompoundObservation': 'CompoundObservation',
+    'Observation': 'Observation',
+    'Kernel': 'Kernel',
+
+}
+
 # -- General configuration ---------------------------------------------------
 
 # Add any Sphinx extension module names here, as strings. They can be