def _initialize_renderer()

in gym_xarm/tasks/base.py [0:0]

• 13 lines of code
• 5 McCabe index (conditional complexity)

    def _initialize_renderer(self, renderer_type: str):
        if renderer_type == "observation":
            model = self.model
        elif renderer_type == "visualization":
            # HACK: gymnasium doesn't allow for custom size rendering on-the-fly, so we
            # initialize another renderer with appropriate size for visualization purposes
            # see https://gymnasium.farama.org/content/migration-guide/#environment-render
            from copy import deepcopy

            model = deepcopy(self.model)
            model.vis.global_.offwidth = self.visualization_width
            model.vis.global_.offheight = self.visualization_height
        else:
            raise ValueError(
                f"Unknown render type {renderer_type}. Must be one of [observation, visualization]"
            )

        return MujocoRenderer(model, self.data)