mirror of
https://github.com/wassname/jaxtyping.git
synced 2026-09-01 12:21:56 +08:00
11 lines
228 B
Python
11 lines
228 B
Python
import cloudpickle
|
|
|
|
from jaxtyping import AbstractArray, Array, Shaped
|
|
|
|
|
|
def test_pickle():
|
|
x = cloudpickle.dumps(Shaped[Array, ""])
|
|
y = cloudpickle.dumps(AbstractArray)
|
|
cloudpickle.loads(x)
|
|
cloudpickle.loads(y)
|