__init__.py 986 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. from __future__ import annotations
  2. from minigrid.envs.babyai.goto import (
  3. GoTo,
  4. GoToDoor,
  5. GoToImpUnlock,
  6. GoToLocal,
  7. GoToObj,
  8. GoToObjDoor,
  9. GoToRedBall,
  10. GoToRedBallGrey,
  11. GoToRedBallNoDists,
  12. GoToRedBlueBall,
  13. GoToSeq,
  14. )
  15. from minigrid.envs.babyai.open import (
  16. Open,
  17. OpenDoor,
  18. OpenDoorsOrder,
  19. OpenRedDoor,
  20. OpenTwoDoors,
  21. )
  22. from minigrid.envs.babyai.other import (
  23. ActionObjDoor,
  24. FindObjS5,
  25. KeyCorridor,
  26. MoveTwoAcross,
  27. OneRoomS8,
  28. )
  29. from minigrid.envs.babyai.pickup import (
  30. Pickup,
  31. PickupAbove,
  32. PickupDist,
  33. PickupLoc,
  34. UnblockPickup,
  35. )
  36. from minigrid.envs.babyai.putnext import PutNext, PutNextLocal
  37. from minigrid.envs.babyai.synth import (
  38. BossLevel,
  39. BossLevelNoUnlock,
  40. MiniBossLevel,
  41. Synth,
  42. SynthLoc,
  43. SynthSeq,
  44. )
  45. from minigrid.envs.babyai.unlock import (
  46. BlockedUnlockPickup,
  47. KeyInBox,
  48. Unlock,
  49. UnlockLocal,
  50. UnlockPickup,
  51. UnlockToUnlock,
  52. )