From 928c3c0dd37bb993d842a909237cbadd22883d03 Mon Sep 17 00:00:00 2001 From: cgerest <148561679+corentingst@users.noreply.github.com> Date: Tue, 5 Mar 2024 23:50:28 +0100 Subject: [PATCH] Update full workflow --- a2c_sb3_panda_reach.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/a2c_sb3_panda_reach.py b/a2c_sb3_panda_reach.py index a2cae91..ca4de87 100644 --- a/a2c_sb3_panda_reach.py +++ b/a2c_sb3_panda_reach.py @@ -1,7 +1,9 @@ import gymnasium as gym + from stable_baselines3 import A2C from stable_baselines3.common.monitor import Monitor from stable_baselines3.common.vec_env import DummyVecEnv + import wandb import panda_gym from wandb.integration.sb3 import WandbCallback -- GitLab