Add missing type hint imports, as suggested by mypy. (#1397)

* Add missing type hint imports, as suggested by mypy.

* Use typing.Callable instead of builtins.callable, which isn't a type.

arviz/data/io_numpyro.py

@@ -1,5 +1,6 @@
 """NumPyro-specific conversion code."""
 import logging
+from typing import Callable, Optional
 
 import numpy as np
 import xarray as xr
@@ -16,7 +17,7 @@ class NumPyroConverter:
 
     # pylint: disable=too-many-instance-attributes
 
-    model = None  # type: Optional[callable]
+    model = None  # type: Optional[Callable]
     nchains = None  # type: int
     ndraws = None  # type: int
 

arviz/data/io_pyro.py

@@ -1,5 +1,6 @@
 """Pyro-specific conversion code."""
 import logging
+from typing import Callable, Optional
 import warnings
 
 import numpy as np
@@ -18,7 +19,7 @@ class PyroConverter:
 
     # pylint: disable=too-many-instance-attributes
 
-    model = None  # type: Optional[callable]
+    model = None  # type: Optional[Callable]
     nchains = None  # type: int
     ndraws = None  # type: int
 

arviz/stats/stats.py

@@ -4,7 +4,7 @@
 import warnings
 from collections import OrderedDict
 from copy import deepcopy
-from typing import List, Optional, Union
+from typing import Any, Dict, List, Optional, Union
 
 import numpy as np
 import pandas as pd