Closed afspies closed 1 year ago
The use of typing such as dict[str, t.Tensor] requires python 3.9 to work. You might consider using Dict[str, Tensor] with e.g. jax-typing, or state Python $>=$ 3.9 as an official req?
dict[str, t.Tensor]
Dict[str, Tensor]
The use of typing such as
dict[str, t.Tensor]
requires python 3.9 to work. You might consider usingDict[str, Tensor]
with e.g. jax-typing, or state Python $>=$ 3.9 as an official req?