native support for fancy_gym
This commit is contained in:
parent
2e7859f068
commit
bd507c37c3
@ -30,7 +30,7 @@ Fancy RL provides two main components:
|
||||
ppo.train()
|
||||
```
|
||||
|
||||
For environments, you can pass any gymnasium environment ID as a string, a function returning a gymnasium environment, or an already instantiated gymnasium environment. Future plans include supporting other torchrl environments.
|
||||
For environments, you can pass any [gymnasium](https://gymnasium.farama.org/) or [Fancy Gym](https://alrhub.github.io/fancy_gym/) environment ID as a string, a function returning a gymnasium environment, or an already instantiated gymnasium environment. Future plans include supporting other torchrl environments.
|
||||
Check 'example/example.py' for a more complete usage example.
|
||||
|
||||
2. **Additional Modules for TRPL**: Designed to integrate with torchrl's primitives-first approach, these modules are ideal for building custom algorithms with precise trust region projections.
|
||||
|
@ -9,6 +9,10 @@ from torchrl.envs import ExplorationType, set_exploration_type
|
||||
from torchrl.envs.libs.gym import GymWrapper
|
||||
from torchrl.record import VideoRecorder
|
||||
import gymnasium as gym
|
||||
try:
|
||||
import fancy_gym
|
||||
except ImportError:
|
||||
pass
|
||||
|
||||
class OnPolicy(ABC):
|
||||
def __init__(
|
||||
|
Loading…
Reference in New Issue
Block a user