readme udpated

fancy_gym/black_box/black_box_wrapper.py

@@ -1,7 +1,3 @@
-import os
-
-os.environ["MUJOCO_GL"] = "egl"
-
 from typing import Tuple, Optional
 
 import gym

fancy_gym/black_box/raw_interface_wrapper.py

@@ -9,10 +9,13 @@ from mp_pytorch.mp.mp_interfaces import MPInterface
 class RawInterfaceWrapper(gym.Wrapper):
 
     @property
-    @abstractmethod
     def context_mask(self) -> np.ndarray:
         """
-        This function defines the contexts. The contexts are defined as specific observations.
+        Returns boolean mask of the same shape as the observation space.
+        It determines whether the observation is returned for the contextual case or not.
+        This effectively allows to filter unwanted or unnecessary observations from the full step-based case.
+        E.g. Velocities starting at 0 are only changing after the first action. Given we only receive the
+        context/part of the first observation, the velocities are not necessary in the observation for the task.
         Returns:
             bool array representing the indices of the observations