From 9c7e6cef3a11c6a795e17154c923eba5e96f39af Mon Sep 17 00:00:00 2001
From: Johannes Fischer <johannes.fischer@kit.edu>
Date: Mon, 13 Sep 2021 16:43:45 +0200
Subject: [PATCH] Rename action to option

---
 scratch/etienne/intersimple/gail_options_image.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scratch/etienne/intersimple/gail_options_image.py b/scratch/etienne/intersimple/gail_options_image.py
index a4afd58..587de9c 100644
--- a/scratch/etienne/intersimple/gail_options_image.py
+++ b/scratch/etienne/intersimple/gail_options_image.py
@@ -154,7 +154,7 @@ def sample(env, generator, discriminator, level: str):
             assert steps > 0
             yield {
                 'obs': obs,
-                'action': ch,
+                'option': ch,
                 'reward': r.detach() / steps,
                 'episode_start': episode_start,
                 'value': value.detach(),
@@ -182,7 +182,7 @@ def train_generator(env, generator, discriminator, num_samples):
     for s in generator_samples[:-1]:
         generator.rollout_buffer.add(
             obs=s['obs'],
-            action=s['action'].cpu(),
+            action=s['option'].cpu(),
             reward=s['reward'].cpu(),
             episode_start=s['episode_start'],
             value=s['value'],