native support for fancy_gym

This commit is contained in:
2024-05-31 13:11:09 +02:00
parent 2e7859f068
commit bd507c37c3
2 changed files with 5 additions and 1 deletions
+4
View File
@@ -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__(