vyeevani

I happened to step away from using Gymnasium APIs. I was focused on Brax. ```python from typing import ClassVar, Optional from brax.envs.base import PipelineEnv from brax.io import image # import...

^ This is really hacky stuff and there's tons that's terrible about it. This is a high level sketch of everything that would be needed to get this to work.

Haven't tested this, but I'm thinking something like this: ``` class AutoResetWrapper2(Wrapper): """Automatically resets Brax envs that are done.""" def reset(self, rng: jax.Array) -> State: base_state = self.env.reset(rng) info =...

This allows us to pass back information from the wrapped things up to the clients that are expecting them (episode wrappers truncated for instance) while at the same time caching...

Tested the above (not very rigorously), seems to work

It’s that except using Gymnasium instead of Gym. The imports would change and also the returns are different for the step API. For example, you’d need to also return truncations...

I should also mention that I tried this on my local machine (arm64 MacBook) plus the binder environment mentioned in the readme

If anyone else has this issue, I ended up working around this with the following: ```python from tqdm import tqdm import numpy as np from stl import mesh from OCC.Core.BRepBuilderAPI...

links to the descriptions: https://developer.apple.com/documentation/coremedia/kcmvideocodectype_422ypcbcr8 https://developer.apple.com/documentation/coremedia/kcmpixelformat_422ypcbcr8_yuvs

Sample code for doing this: ``` import os import shutil from absl import app, flags import tqdm import glob import multiprocessing """ Converts from the following tree structure to bridgedata_raw/...