def testNumValues(self, dtype): spec = array_spec.BoundedArraySpec(tuple(), dtype, minimum=0, maximum=9) num_values = spec.num_values if array_spec.is_discrete(spec): self.assertEqual(10, num_values) else: self.assertEqual(None, num_values)
def is_discrete(spec): if isinstance(spec, TensorSpec): return spec.dtype.is_integer else: return array_spec.is_discrete(spec)