[WIP] fix mypy errors (arviz-devs#1586)

* Changed type ints for var_names * fix mypy errors
utkarsh-maheshwari · May 27, 2021 · dc4a6f1 · dc4a6f1
1 parent 58747e5
commit dc4a6f1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/arviz/plots/traceplot.py b/arviz/plots/traceplot.py
@@ -1,6 +1,6 @@
 """Plot kde or histograms and values from MCMC samples."""
 import warnings
-from typing import Any, Callable, List, Mapping, Optional, Tuple, Union
+from typing import Any, Callable, List, Mapping, Optional, Tuple, Union, Sequence
 
 from ..data import CoordSpec, InferenceData, convert_to_dataset
 from ..labels import BaseLabeller
@@ -12,7 +12,7 @@
 
 def plot_trace(
     data: InferenceData,
-    var_names: Optional[List[str]] = None,
+    var_names: Optional[Sequence[str]] = None,
     filter_vars: Optional[str] = None,
     transform: Optional[Callable] = None,
     coords: Optional[CoordSpec] = None,

diff --git a/doc/source/conf.py b/doc/source/conf.py
@@ -20,6 +20,7 @@
 import os
 import re
 import sys
+from typing import Dict
 
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
 import arviz
@@ -181,7 +182,7 @@
 
 # -- Options for LaTeX output ---------------------------------------------
 
-latex_elements = {
+latex_elements : Dict[str, str] = {
     # The paper size ('letterpaper' or 'a4paper').
     #
     # 'papersize': 'letterpaper',