added policy_adaption

2021-06-04 18:13:55 +02:00
parent b70b12edda
commit 669762a453
2 changed files with 3 additions and 23 deletions
--- a/environments/policy_adaption/natural_rl_environment/natural_env.py
+++ b/environments/policy_adaption/natural_rl_environment/natural_env.py
@ -69,7 +69,7 @@ def make(name='Pong-v0', imgsource='color', files=None):
        imgsource = RandomColorSource(shape2d)
    elif imgsource == "noise":
        imgsource = NoiseSource(shape2d)
-    if args.imgsource == "images":
+    elif imgsource == "images":
        imgsource = RandomImageSource(shape2d, files)
    else:
        raise NotImplementedError(f'{imgsource} is not supported, use one of {{video, color, noise}}')