How to use the ptype.ArrayT function in ptype
To help you get started, we’ve selected a few ptype examples, based on popular ways it is used in public projects.
Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
iskandr / parakeet / ptype.py View on Github 
def __eq__(self, other):
return isinstance(other, ArrayT) and \
self.elt_type == other.elt_type and self.rank == other.rankdef type_of_value(x):
if np.isscalar(x):
return type_of_scalar(x)
elif isinstance(x, tuple):
elt_types = map(type_of_value, x)
return TupleT(elt_types)
elif isinstance(x, np.ndarray):
return ArrayT(from_dtype(x.dtype), np.rank(x))
else:
raise RuntimeError("Unsupported type " + str(type(x)))
