def testNoSpecMissingOrColliding(self): spec_names = set() flat_spec = tf.nest.flatten(self._time_step_spec) missing_or_colliding = [ policy_saver._true_if_missing_or_collision(s, spec_names) for s in flat_spec ] self.assertFalse(any(missing_or_colliding))
def testTrueIfMissing(self): time_step_spec = self._time_step_spec._replace( observation=tensor_spec.BoundedTensorSpec( dtype=tf.float32, shape=(4, ), minimum=-10.0, maximum=10.0)) spec_names = set() flat_spec = tf.nest.flatten(time_step_spec) missing_or_colliding = [ policy_saver._true_if_missing_or_collision(s, spec_names) for s in flat_spec ] self.assertTrue(any(missing_or_colliding))