wassname/jaxtyping
1
0
Fork 0
mirror of https://github.com/wassname/jaxtyping.git synced 2026-09-24 13:30:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed cloudpickle breaking, mark 2

Browse Source
This commit is contained in:
Patrick Kidger
2023-03-02 17:37:53 -08:00
parent 2b339715f9
commit 5600a1aac8
3 changed files with 95 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_serialisation.py
+4 -2
View File
@@ -4,5 +4,7 @@ from jaxtyping import AbstractArray, Array, Shaped
def test_pickle():
cloudpickle.dumps(Shaped[Array, ""])
cloudpickle.dumps(AbstractArray)
x = cloudpickle.dumps(Shaped[Array, ""])
y = cloudpickle.dumps(AbstractArray)
cloudpickle.loads(x)
cloudpickle.loads(y)