Fix image_shape key error (#58)

rail-berkeley · Apr 3, 2019 · 1f6147c · 1f6147c
1 parent 1f291b2
commit 1f6147c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/development/variants.py b/examples/development/variants.py
@@ -286,8 +286,9 @@ def get_variant_spec_image(universe,
             'kwargs': {
                 'image_shape': (
                     variant_spec
-                    ['training']
                     ['environment_params']
+                    ['training']
+                    ['kwargs']
                     ['image_shape']),
                 'output_size': M,
                 'conv_filters': (4, 4),