From 7016f57fa2a1143b1e10a1da3c2967263f8df30e Mon Sep 17 00:00:00 2001 From: Lorenzo Bonanni Date: Tue, 4 Mar 2025 14:49:41 +0100 Subject: [PATCH] Update train_script.py Fixed `gymnasium.error.NameNotFound: Environment `MiniGrid-DoorKey-5x5` doesn't exist.` --- sr_dqn_lab/train_script.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sr_dqn_lab/train_script.py b/sr_dqn_lab/train_script.py index cab2e6a..e03f302 100644 --- a/sr_dqn_lab/train_script.py +++ b/sr_dqn_lab/train_script.py @@ -1,6 +1,7 @@ import os import gymnasium as gym import numpy as np +import minigrid import argparse import matplotlib.pyplot as plt from stable_baselines3 import DQN