fixup!82604: set dtype correctly

Author: Kakadus

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,9 +74,9 @@ def _create_data(self, dataContainerPb):
             # print(boxContainerPb.shape, boxContainerPb.dtype, boxContainerPb.uintData)
 
             if boxContainerPb.dtype == pb.INT:
-                data = np.array(boxContainerPb.doubleData, dtype=int)
+                data = np.array(boxContainerPb.intData, dtype=int)
             elif boxContainerPb.dtype == pb.UINT:
-                data = np.array(boxContainerPb.doubleData, dtype=np.uint)
+                data = np.array(boxContainerPb.uintData, dtype=np.uint)
             elif boxContainerPb.dtype == pb.DOUBLE:
                 data = np.array(boxContainerPb.doubleData, dtype=np.float64)
             else: