Added None check to moma_option.py.

PiperOrigin-RevId: 704265704 Change-Id: Icfb01e9bee501cee22bec887673a8d88689e027e
google-deepmind · Dec 9, 2024 · 99d3dc5 · 99d3dc5
1 parent 8c195bb
commit 99d3dc5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/py/moma/moma_option.py b/py/moma/moma_option.py
@@ -78,6 +78,9 @@ def _set_control(self, physics: mjcf.Physics, command: np.ndarray):
       # charge of controlling individual effectors.
       return
 
+    if self._action_spec is None:
+      raise ValueError('self._action_spec is None.')
+
     spec_utils.validate(self._action_spec, command)
 
     for ef in self._effectors: