From c5b043c49ffa1b8b05c4db792f04b09d81fb2720 Mon Sep 17 00:00:00 2001 From: Johannes Fischer Date: Thu, 28 Oct 2021 13:47:54 +0200 Subject: [PATCH] Cleanup --- scratch/johannes/gail_options_image.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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