def testDatasetAsNumpy(self): arrs = extensions.dataset_as_numpy( [tf.constant([1, 2]), tf.constant([3, 4])]) for a in arrs: self.assertIsInstance(a, arrays.ndarray) with self.assertRaisesWithPredicateMatch( ValueError, r"dataset_as_numpy must be run in eager mode outside tf.function"): @tf.function def f(): return extensions.dataset_as_numpy([tf.constant([1, 2])]) f()
def f(): return extensions.dataset_as_numpy([tf.constant([1, 2])])