Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Merged
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
3 changes: 2 additions & 1 deletion 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
Expand All @@ -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

Expand Down
3 changes: 2 additions & 1 deletion 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
Expand All @@ -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

Expand Down
2 changes: 1 addition & 1 deletion arviz/stats/stats.py
Expand Up @@ -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
Expand Down