removed imagerecordercallback

9 months ago · ed0fc6134e
2 changed files with 2 additions and 2 deletions
--- a/notebooks/DeterministicShielding.ipynb
+++ b/notebooks/DeterministicShielding.ipynb
@ -55,7 +55,7 @@
    "import os\n",
    "\n",
    "from utils import MiniGridShieldHandler, create_log_dir, ShieldingConfig, MiniWrapper, expname, shield_needed, shielded_evaluation, create_shield_overlay_image\n",
-    "from sb3utils import MiniGridSbShieldingWrapper, parse_sb3_arguments, ImageRecorderCallback, InfoCallback\n",
+    "from sb3utils import MiniGridSbShieldingWrapper, parse_sb3_arguments, InfoCallback\n",
    "\n",
    "import os, sys\n",
    "from copy import deepcopy\n",
--- a/notebooks/ProbabilisticShielding.ipynb
+++ b/notebooks/ProbabilisticShielding.ipynb
@ -55,7 +55,7 @@
    "import os\n",
    "\n",
    "from utils import MiniGridShieldHandler, create_log_dir, ShieldingConfig, MiniWrapper, expname, shield_needed, shielded_evaluation, create_shield_overlay_image\n",
-    "from sb3utils import MiniGridSbShieldingWrapper, parse_sb3_arguments, ImageRecorderCallback, InfoCallback\n",
+    "from sb3utils import MiniGridSbShieldingWrapper, parse_sb3_arguments, InfoCallback\n",
    "\n",
    "import os, sys\n",
    "from copy import deepcopy\n",