d4rl_env.py 835 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. """
  2. 8 Environments from D4RL Environment.
  3. Use fully qualified class-path in your configs:
  4. e.g. "env": "ray.rllib.examples.env.d4rl_env.halfcheetah_random".
  5. """
  6. import gym
  7. try:
  8. import d4rl
  9. d4rl.__name__ # Fool LINTer.
  10. except ImportError:
  11. d4rl = None
  12. def halfcheetah_random():
  13. return gym.make("halfcheetah-random-v0")
  14. def halfcheetah_medium():
  15. return gym.make("halfcheetah-medium-v0")
  16. def halfcheetah_expert():
  17. return gym.make("halfcheetah-expert-v0")
  18. def halfcheetah_medium_replay():
  19. return gym.make("halfcheetah-medium-replay-v0")
  20. def hopper_random():
  21. return gym.make("hopper-random-v0")
  22. def hopper_medium():
  23. return gym.make("hopper-medium-v0")
  24. def hopper_expert():
  25. return gym.make("hopper-expert-v0")
  26. def hopper_medium_replay():
  27. return gym.make("hopper-medium-replay-v0")