- Installing PyTorch and flash_attn
- Patching Industrial Benchmark
IDS.py
file in order to be compatible withnumpy>=1.24.4
- Patching and installing Industrial Benchmark
In newly created environment run these commands:
from envs.wrappers.industrial_benchmark import get_gym_env, TASKS
print(f"Available tasks are:", *TASKS, sep='\n')
# Initialize env
env = get_gym_env(task_name="industrial-benchmark-0-v1")