Commit fc744dae authored by Nicola Gatto's avatar Nicola Gatto
Browse files

Reshape environment state for multidimensional states

parent 3366990b
Pipeline #149570 failed with stages
......@@ -212,7 +212,7 @@ class RosEnvironment(Environment):
rospy.signal_shutdown('Program ended!')
def __state_callback(self, data):
self.__last_received_state = np.array(data.data, dtype='float32')
self.__last_received_state = np.array(data.data, dtype='float32').reshape((<#list config.stateDim as d>${d},</#list>))
rospy.logdebug('Received state: {}'.format(self.__last_received_state))
self.__waiting_for_state_update = False
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment