diff --git a/pymc_extras/model/marginal/graph_analysis.py b/pymc_extras/model/marginal/graph_analysis.py index 6a7a7f874..8ca1ba9ef 100644 --- a/pymc_extras/model/marginal/graph_analysis.py +++ b/pymc_extras/model/marginal/graph_analysis.py @@ -7,7 +7,7 @@ from pymc.model.fgraph import ModelVar from pymc.variational.minibatch_rv import MinibatchRandomVariable from pytensor.graph import Variable, ancestors -from pytensor.graph.basic import io_toposort +from pytensor.graph.traversal import io_toposort from pytensor.tensor import TensorType, TensorVariable from pytensor.tensor.blockwise import Blockwise from pytensor.tensor.elemwise import CAReduce, DimShuffle, Elemwise diff --git a/pymc_extras/test.txt b/pymc_extras/test.txt new file mode 100644 index 000000000..548b06a77 --- /dev/null +++ b/pymc_extras/test.txt @@ -0,0 +1 @@ +THIS IS A TEST TO SEE IF CERTAIN FAILURES ARE RELATED TO A SEPERATE PR OR NOT. PLEASE DISREGARD diff --git a/tests/statespace/core/test_statespace.py b/tests/statespace/core/test_statespace.py index 3895a1a2d..bf27868b4 100644 --- a/tests/statespace/core/test_statespace.py +++ b/tests/statespace/core/test_statespace.py @@ -14,7 +14,7 @@ from numpy.testing import assert_allclose from pymc.testing import mock_sample_setup_and_teardown from pytensor.compile import SharedVariable -from pytensor.graph.basic import graph_inputs +from pytensor.graph.traversal import graph_inputs from pymc_extras.statespace.core.statespace import FILTER_FACTORY, PyMCStateSpace from pymc_extras.statespace.models import structural as st diff --git a/tests/statespace/models/structural/components/test_autoregressive.py b/tests/statespace/models/structural/components/test_autoregressive.py index 33758f0d4..fb2ba87e0 100644 --- a/tests/statespace/models/structural/components/test_autoregressive.py +++ b/tests/statespace/models/structural/components/test_autoregressive.py @@ -4,7 +4,7 @@ from numpy.testing import assert_allclose from pytensor import config -from pytensor.graph.basic import explicit_graph_inputs +from pytensor.graph.traversal import explicit_graph_inputs from pymc_extras.statespace.models import structural as st from tests.statespace.models.structural.conftest import _assert_basic_coords_correct diff --git a/tests/statespace/models/structural/components/test_cycle.py b/tests/statespace/models/structural/components/test_cycle.py index 2371ab597..f30d113c2 100644 --- a/tests/statespace/models/structural/components/test_cycle.py +++ b/tests/statespace/models/structural/components/test_cycle.py @@ -3,7 +3,7 @@ from numpy.testing import assert_allclose from pytensor import config -from pytensor.graph.basic import explicit_graph_inputs +from pytensor.graph.traversal import explicit_graph_inputs from scipy import linalg from pymc_extras.statespace.models import structural as st diff --git a/tests/statespace/models/structural/components/test_measurement_error.py b/tests/statespace/models/structural/components/test_measurement_error.py index a20d56ad7..3bfca5840 100644 --- a/tests/statespace/models/structural/components/test_measurement_error.py +++ b/tests/statespace/models/structural/components/test_measurement_error.py @@ -1,7 +1,7 @@ import numpy as np import pytensor -from pytensor.graph.basic import explicit_graph_inputs +from pytensor.graph.traversal import explicit_graph_inputs from pymc_extras.statespace.models import structural as st from tests.statespace.models.structural.conftest import _assert_basic_coords_correct diff --git a/tests/statespace/models/structural/components/test_seasonality.py b/tests/statespace/models/structural/components/test_seasonality.py index 353ccbe24..22b700174 100644 --- a/tests/statespace/models/structural/components/test_seasonality.py +++ b/tests/statespace/models/structural/components/test_seasonality.py @@ -3,7 +3,7 @@ import pytest from pytensor import config -from pytensor.graph.basic import explicit_graph_inputs +from pytensor.graph.traversal import explicit_graph_inputs from pymc_extras.statespace.models import structural as st from pymc_extras.statespace.models.structural.components.seasonality import FrequencySeasonality diff --git a/tests/statespace/models/test_DFM.py b/tests/statespace/models/test_DFM.py index 81f82d2c3..8294c3aaf 100644 --- a/tests/statespace/models/test_DFM.py +++ b/tests/statespace/models/test_DFM.py @@ -9,7 +9,7 @@ import statsmodels.api as sm from numpy.testing import assert_allclose -from pytensor.graph.basic import explicit_graph_inputs +from pytensor.graph.traversal import explicit_graph_inputs from statsmodels.tsa.statespace.dynamic_factor import DynamicFactor from pymc_extras.statespace.models.DFM import BayesianDynamicFactor diff --git a/tests/statespace/models/test_ETS.py b/tests/statespace/models/test_ETS.py index 5ef1f8c0b..6e2c63c52 100644 --- a/tests/statespace/models/test_ETS.py +++ b/tests/statespace/models/test_ETS.py @@ -4,7 +4,7 @@ import statsmodels.api as sm from numpy.testing import assert_allclose -from pytensor.graph.basic import explicit_graph_inputs +from pytensor.graph.traversal import explicit_graph_inputs from scipy import linalg from pymc_extras.statespace.models.ETS import BayesianETS diff --git a/tests/statespace/models/test_SARIMAX.py b/tests/statespace/models/test_SARIMAX.py index 0daf1c0a5..1d7a81c86 100644 --- a/tests/statespace/models/test_SARIMAX.py +++ b/tests/statespace/models/test_SARIMAX.py @@ -10,7 +10,7 @@ from numpy.testing import assert_allclose, assert_array_less from pymc.testing import mock_sample_setup_and_teardown -from pytensor.graph.basic import explicit_graph_inputs +from pytensor.graph.traversal import explicit_graph_inputs from pymc_extras.statespace import BayesianSARIMAX from pymc_extras.statespace.models.utilities import (