diff --git a/scratch/johannes/gail_options_image.py b/scratch/johannes/gail_options_image.py
index 03849d5..1d84807 100644
--- a/scratch/johannes/gail_options_image.py
+++ b/scratch/johannes/gail_options_image.py
@@ -5,10 +5,9 @@
 from src.discriminator import CnnDiscriminator, CnnDiscriminatorFlatAction
 from imitation.algorithms import adversarial
 import stable_baselines3
-from stable_baselines3.common.evaluation import evaluate_policy
 import torch.utils.data
 import numpy as np
-from intersim.envs.intersimple import Intersimple, NRasterized, NRasterizedInfo, NRasterizedIncrementingAgent, NRasterizedRandomAgent, speed_reward
+from intersim.envs.intersimple import NRasterized, NRasterizedRandomAgent, speed_reward
 import itertools
 import functools
 from torch.distributions import Categorical