code: def add_keyup_callback(self, key, fn):
          
          self.viewer.keyup[key].append(fn)
docs: Allows for custom callback functions for the viewer. Called on key up.
          Parameter 'any' will ensure that the callback is called on any key up,
          and block default mujoco viewer callbacks from executing, except for 
          the ESC callback to close the viewer.