I am using python3.9 and I am getting the following unit test error when installing:
Traceback (most recent call last):
File "/home/jpilault/jaxpruner/jaxpruner/sparsity_distributions_test.py", line 63, in testUniformSparsityMapGeneratorWithCustomMap
result = sparsity_distributions.uniform(
File "/home/jpilault/jaxpruner/jaxpruner/sparsity_distributions.py", line 72, in uniform
if isinstance(params, chex.Array):
File "/usr/lib/python3.9/typing.py", line 703, in __instancecheck__
return self.__subclasscheck__(type(obj))
File "/usr/lib/python3.9/typing.py", line 706, in __subclasscheck__
raise TypeError("Subscripted generics cannot be used with"
TypeError: Subscripted generics cannot be used with class and instance checks
I am using python3.9 and I am getting the following unit test error when installing: