Minor fix with an assertion

jscriptcoder · jscriptcoder · commit 0140f3ba939e · 2020-10-14T22:11:58.000+02:00
diff --git a/gymie/__init__.py b/gymie/__init__.py
@@ -2,4 +2,4 @@
 from gymie.api import override
 from gymie.exceptions import *
 
-__version__ = '0.1.0'
+__version__ = '0.1.1'
diff --git a/gymie/api.py b/gymie/api.py
@@ -40,14 +40,11 @@ def override(func_name):
     Raises:
         AssertionError: wrong function to override
     """
-
+    assert func_name in ['get_env', 'process_step'], \
+        'Error overriding. Functions available: `get_env`, `process_step`'
+    
     def inner_override(fn):
-        assert(
-            func_name in ['get_env', 'process_step'], 
-            'Error overriding. Functions available: `get_env`, `process_step`')
-
         globals()[func_name] = fn
-
         return fn
     
     return inner_override
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="gymie",
-    version="0.1.0",
+    version="0.1.1",
     author="Francisco Ramos",
     author_email="francisco.ramos@researchlab.ai",
     description="WebSocket server that exposes an API to train AI agents on OpenAI Gym and gym-api-like Environments",
diff --git a/tests/unity_env/PushBlock.app/Contents/ML-Agents/Timers/PushBlock_timers.json b/tests/unity_env/PushBlock.app/Contents/ML-Agents/Timers/PushBlock_timers.json
@@ -1 +1 @@
-{"count":1,"self":0.59151999999999993,"total":0.65144,"children":{"InitializeActuators":{"count":1,"self":0.0028049999999999998,"total":0.0028049999999999998,"children":null},"InitializeSensors":{"count":1,"self":0.002598,"total":0.002598,"children":null},"AgentSendState":{"count":10,"self":0.0024939999999999997,"total":0.015099,"children":{"CollectObservations":{"count":4,"self":3.7E-05,"total":3.7E-05,"children":null},"CollectDiscreteActionMasks":{"count":4,"self":0.00033,"total":0.00033,"children":null},"RequestDecision":{"count":4,"self":0.00328,"total":0.012237999999999999,"children":{"AgentInfo.ToProto":{"count":4,"self":0.000993,"total":0.0089579999999999989,"children":{"GenerateSensorData":{"count":4,"self":0.0038859999999999997,"total":0.007965,"children":{"RayPerceptionSensor.Perceive":{"count":8,"self":0.004079,"total":0.004079,"children":null}}}}}}}}},"DecideAction":{"count":10,"self":0.035276999999999996,"total":0.035276999999999996,"children":null},"AgentAct":{"count":10,"self":0.001748,"total":0.001748,"children":null}},"gauges":{},"metadata":{"timer_format_version":"0.1.0","start_time_seconds":"1602684683","unity_version":"2019.4.0f1","command_line_arguments":"\/Users\/franciscoramos\/Personal\/Gymie-Server\/tests\/unity_env\/PushBlock.app\/Contents\/MacOS\/UnityEnvironment --mlagents-port 5005","communication_protocol_version":"1.1.0","com.unity.ml-agents_version":"1.4.0-preview","scene_name":"PushBlock","end_time_seconds":"1602684683"}}
+{"count":1,"self":20360.5835776,"total":20411.081261,"children":{"InitializeActuators":{"count":1,"self":0.002441,"total":0.002441,"children":null},"InitializeSensors":{"count":1,"self":0.0026579999999999998,"total":0.0026579999999999998,"children":null},"AgentSendState":{"count":480052,"self":9.5747552,"total":25.378849,"children":{"CollectObservations":{"count":96011,"self":0.486523,"total":0.486523,"children":null},"CollectDiscreteActionMasks":{"count":96011,"self":0.55143599999999993,"total":0.55143599999999993,"children":null},"RequestDecision":{"count":96011,"self":3.103964,"total":14.766134999999998,"children":{"RayPerceptionSensor.Perceive":{"count":192022,"self":11.6621712,"total":11.662170999999999,"children":null}}}}},"DecideAction":{"count":480052,"self":19.3603296,"total":19.360329999999998,"children":null},"AgentAct":{"count":480052,"self":5.7394472,"total":5.750506,"children":{"RayPerceptionSensor.Perceive":{"count":192,"self":0.011059,"total":0.011059,"children":null}}}},"gauges":{"PushBlock.CumulativeReward":{"count":96,"max":-0.99994123,"min":-0.99994123,"runningAverage":-0.99994123,"value":-0.99994123,"weightedAverage":-0.99994123}},"metadata":{"timer_format_version":"0.1.0","start_time_seconds":"1602684619","unity_version":"2019.4.0f1","command_line_arguments":"\/Users\/franciscoramos\/Personal\/Gymie-Server\/tests\/unity_env\/PushBlock.app\/Contents\/MacOS\/UnityEnvironment --mlagents-port 5005","communication_protocol_version":"1.1.0","com.unity.ml-agents_version":"1.4.0-preview","scene_name":"PushBlock","end_time_seconds":"1602705030"}}

Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		-{"count":1,"self":0.59151999999999993,"total":0.65144,"children":{"InitializeActuators":{"count":1,"self":0.0028049999999999998,"total":0.0028049999999999998,"children":null},"InitializeSensors":{"count":1,"self":0.002598,"total":0.002598,"children":null},"AgentSendState":{"count":10,"self":0.0024939999999999997,"total":0.015099,"children":{"CollectObservations":{"count":4,"self":3.7E-05,"total":3.7E-05,"children":null},"CollectDiscreteActionMasks":{"count":4,"self":0.00033,"total":0.00033,"children":null},"RequestDecision":{"count":4,"self":0.00328,"total":0.012237999999999999,"children":{"AgentInfo.ToProto":{"count":4,"self":0.000993,"total":0.0089579999999999989,"children":{"GenerateSensorData":{"count":4,"self":0.0038859999999999997,"total":0.007965,"children":{"RayPerceptionSensor.Perceive":{"count":8,"self":0.004079,"total":0.004079,"children":null}}}}}}}}},"DecideAction":{"count":10,"self":0.035276999999999996,"total":0.035276999999999996,"children":null},"AgentAct":{"count":10,"self":0.001748,"total":0.001748,"children":null}},"gauges":{},"metadata":{"timer_format_version":"0.1.0","start_time_seconds":"1602684683","unity_version":"2019.4.0f1","command_line_arguments":"\/Users\/franciscoramos\/Personal\/Gymie-Server\/tests\/unity_env\/PushBlock.app\/Contents\/MacOS\/UnityEnvironment --mlagents-port 5005","communication_protocol_version":"1.1.0","com.unity.ml-agents_version":"1.4.0-preview","scene_name":"PushBlock","end_time_seconds":"1602684683"}}
	`1`	+{"count":1,"self":20360.5835776,"total":20411.081261,"children":{"InitializeActuators":{"count":1,"self":0.002441,"total":0.002441,"children":null},"InitializeSensors":{"count":1,"self":0.0026579999999999998,"total":0.0026579999999999998,"children":null},"AgentSendState":{"count":480052,"self":9.5747552,"total":25.378849,"children":{"CollectObservations":{"count":96011,"self":0.486523,"total":0.486523,"children":null},"CollectDiscreteActionMasks":{"count":96011,"self":0.55143599999999993,"total":0.55143599999999993,"children":null},"RequestDecision":{"count":96011,"self":3.103964,"total":14.766134999999998,"children":{"RayPerceptionSensor.Perceive":{"count":192022,"self":11.6621712,"total":11.662170999999999,"children":null}}}}},"DecideAction":{"count":480052,"self":19.3603296,"total":19.360329999999998,"children":null},"AgentAct":{"count":480052,"self":5.7394472,"total":5.750506,"children":{"RayPerceptionSensor.Perceive":{"count":192,"self":0.011059,"total":0.011059,"children":null}}}},"gauges":{"PushBlock.CumulativeReward":{"count":96,"max":-0.99994123,"min":-0.99994123,"runningAverage":-0.99994123,"value":-0.99994123,"weightedAverage":-0.99994123}},"metadata":{"timer_format_version":"0.1.0","start_time_seconds":"1602684619","unity_version":"2019.4.0f1","command_line_arguments":"\/Users\/franciscoramos\/Personal\/Gymie-Server\/tests\/unity_env\/PushBlock.app\/Contents\/MacOS\/UnityEnvironment --mlagents-port 5005","communication_protocol_version":"1.1.0","com.unity.ml-agents_version":"1.4.0-preview","scene_name":"PushBlock","end_time_seconds":"1602705030"}}