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))
