diff --git a/setup.py b/setup.py index f767df6ae9..00ac659b6c 100644 --- a/setup.py +++ b/setup.py @@ -39,7 +39,7 @@ python_requires="~=3.7", install_requires=[ "absl-py", - "dill!=0.3.6", + "dill<0.3.6", "gpflow>=2.8.1", "gpflux>=0.4.2", "numpy", diff --git a/tests/latest/constraints.txt b/tests/latest/constraints.txt index be946e9d2a..be6b0fa4b3 100644 --- a/tests/latest/constraints.txt +++ b/tests/latest/constraints.txt @@ -15,7 +15,7 @@ cvxpy==1.4.2 cycler==0.12.1 decorator==5.1.1 Deprecated==1.2.14 -dill==0.3.8 +dill==0.3.5.1 dm-tree==0.1.8 dropstackframe==0.1.0 ecos==2.0.13 @@ -51,7 +51,7 @@ osqp==0.6.5 packaging==23.2 pillow==10.2.0 pluggy==1.4.0 -protobuf==4.23.4 +protobuf==4.25.2 pyasn1==0.5.1 pyasn1-modules==0.3.0 pybind11==2.11.1 @@ -68,7 +68,7 @@ scipy==1.11.4 scs==3.2.4.post1 six==1.16.0 tabulate==0.9.0 -tensorboard==2.15.1 +tensorboard==2.15.2 tensorboard-data-server==0.7.2 tensorflow==2.15.0 tensorflow-estimator==2.15.0 diff --git a/tests/unit/test_bayesian_optimizer.py b/tests/unit/test_bayesian_optimizer.py index fce0a19343..54362625ce 100644 --- a/tests/unit/test_bayesian_optimizer.py +++ b/tests/unit/test_bayesian_optimizer.py @@ -21,9 +21,7 @@ import numpy.testing as npt import pytest import tensorflow as tf -import tensorflow_probability as tfp from check_shapes import inherit_check_shapes -from packaging.version import Version from tests.unit.test_ask_tell_optimization import DatasetChecker, LocalDatasetsFixedAcquisitionRule from tests.util.misc import ( @@ -667,10 +665,6 @@ def optimize(self, dataset: Dataset) -> None: @pytest.mark.parametrize("save_to_disk", [False, True]) def test_bayesian_optimizer_optimize_tracked_state(save_to_disk: bool) -> None: - if save_to_disk and Version(tfp.__version__) >= Version("0.23.0"): - # TODO: the latest tfp seems to have broken pickling QuadraticMeanAndRBFKernel - pytest.skip() - class _CountingRule(AcquisitionRule[State[Optional[int], TensorType], Box, ProbabilisticModel]): def acquire( self,