Fixed action handling in agents

programlib/agent.py

@@ -9,9 +9,12 @@ def decode_action(action):
         except SyntaxError:
             x = np.array(list(map(eval, action.split(r'[\p\s]+'))))
 
-        return x.reshape(-1)
+        x = x.reshape(-1)
+        if x.size == 1:
+            x = x[0]
+        return x
     except SyntaxError as e:
-        raise RuntimeError(f'Action {action} not recognized. Please use numbers only') from e
+        raise RuntimeError(f'{action}\nis not a valid action') from e
 
 def encode_obs(obs):
     try:

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "programlib"
-version = "10.3.1"
+version = "10.4.0"
 description = "Programs as Objects"
 authors = ["Vadim Liventsev <[email protected]>"]
 license = "MIT"