Fix imports in 3_train_gail.ipynb

HumanCompatibleAI · Aug 8, 2023 · 50e5970 · 50e5970
1 parent c5600cf
commit 50e5970
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/tutorials/3_train_gail.ipynb b/docs/tutorials/3_train_gail.ipynb
@@ -89,7 +89,7 @@
    "outputs": [],
    "source": [
     "from imitation.algorithms.adversarial.gail import GAIL\n",
-    "from imitation.rewards.reward_nets import BasicRewardNet, BasicShapedRewardNet\n",
+    "from imitation.rewards.reward_nets import BasicShapedRewardNet\n",
     "from imitation.util.networks import RunningNorm\n",
     "from imitation.util.util import make_vec_env\n",
     "from stable_baselines3 import PPO\n",