Fixed cloudpickle breaking

This commit is contained in:
Patrick Kidger
2023-03-02 12:35:38 -08:00
parent 8c86958b77
commit 2b339715f9
3 changed files with 27 additions and 12 deletions
+1
View File
@@ -2,3 +2,4 @@ equinox>=0.5.3
pytest>=7.0.1
beartype>=0.10.4
typeguard>=2.13.3
cloudpickle>=2.2.1
+8
View File
@@ -0,0 +1,8 @@
import cloudpickle
from jaxtyping import AbstractArray, Array, Shaped
def test_pickle():
cloudpickle.dumps(Shaped[Array, ""])
cloudpickle.dumps(AbstractArray)