fix: check and set dtype of box correctly

Author: OliverZim

model/gym-interface/py/ns3ai_gym_env/envs/ns3_environment.py

@@ -61,6 +61,7 @@ def _create_space(self, spaceDesc):
         return space
 
     def _create_data(self, dataContainerPb):
+        import pdb; pdb.set_trace()
         if dataContainerPb.type == pb.Discrete:
             discreteContainerPb = pb.DiscreteDataContainer()
             dataContainerPb.data.Unpack(discreteContainerPb)
@@ -73,16 +74,15 @@ def _create_data(self, dataContainerPb):
             # print(boxContainerPb.shape, boxContainerPb.dtype, boxContainerPb.uintData)
 
             if boxContainerPb.dtype == pb.INT:
-                data = boxContainerPb.intData
+                data = np.array(boxContainerPb.intData, dtype=int)
             elif boxContainerPb.dtype == pb.UINT:
-                data = boxContainerPb.uintData
+                data = np.array(boxContainerPb.uintData, dtype=np.uint)
             elif boxContainerPb.dtype == pb.DOUBLE:
-                data = boxContainerPb.doubleData
+                data = np.array(boxContainerPb.doubleData, dtype=np.float64)
             else:
-                data = boxContainerPb.floatData
+                data = np.array(boxContainerPb.floatData, dtype=np.float32)
 
             # TODO: reshape using shape info
-            data = np.array(data)
             return data
 
         elif dataContainerPb.type == pb.Tuple:
@@ -207,11 +207,11 @@ def _pack_data(cls, actions, spaceDesc):
                 boxContainerPb.dtype = pb.UINT
                 boxContainerPb.uintData.extend(actions)
 
-            elif spaceDesc.dtype in ['float', 'float32', 'float64']:
+            elif spaceDesc.dtype.name in ["float", "float32"]:
                 boxContainerPb.dtype = pb.FLOAT
                 boxContainerPb.floatData.extend(actions)
 
-            elif spaceDesc.dtype in ['double']:
+            elif spaceDesc.dtype.name in ["double", "float64"]:
                 boxContainerPb.dtype = pb.DOUBLE
                 boxContainerPb.doubleData.extend(actions)