Import missing constants

Author: fonnesbeck

tests/__init__.py

@@ -11,6 +11,3 @@
 #   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
-
-from pymc_extras.distributions import histogram_utils
-from pymc_extras.distributions.histogram_utils import histogram_approximation

tests/distributions/test_discrete_markov_chain.py

@@ -12,7 +12,7 @@
 
 from pymc_extras.distributions.timeseries import (
     DiscreteMarkovChain,
-    DiscreteMarkovChainRV,
+    DiscreteMarkovChainGibbsMetropolis,
 )
 
 

tests/statespace/test_kalman_filter.py

@@ -7,6 +7,7 @@
 
 from pymc_extras.statespace.filters import (
     CholeskyFilter,
+    KalmanSmoother,
     SingleTimeseriesFilter,
     StandardFilter,
     SteadyStateFilter,

tests/statespace/test_statespace.py

@@ -13,10 +13,15 @@
 from pymc_extras.statespace.models import structural as st
 from pymc_extras.statespace.models.utilities import make_default_coords
 from pymc_extras.statespace.utils.constants import (
+    FILTER_OUTPUT_NAMES,
     JITTER_DEFAULT,
     LONG_MATRIX_NAMES,
+    MATRIX_NAMES,
     MISSING_FILL,
+    NEVER_TIME_VARYING,
     SHORT_NAME_TO_LONG,
+    SMOOTHER_OUTPUT_NAMES,
+    VECTOR_VALUED,
 )
 from tests.statespace.utilities.shared_fixtures import (
     rng,

tests/statespace/test_structural.py

@@ -17,9 +17,16 @@
 
 from pymc_extras.statespace import structural as st
 from pymc_extras.statespace.utils.constants import (
+    ALL_STATE_AUX_DIM,
+    ALL_STATE_DIM,
+    AR_PARAM_DIM,
     JITTER_DEFAULT,
     LONG_MATRIX_NAMES,
     MISSING_FILL,
+    OBS_STATE_AUX_DIM,
+    OBS_STATE_DIM,
+    SHOCK_AUX_DIM,
+    SHOCK_DIM,
     SHORT_NAME_TO_LONG,
 )
 from tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import