Fixing bugs and started integration

com.unity.ml-agents/Runtime/Communicator/GrpcExtensions.cs

                 {
                     observationProto.CompressedChannelMapping.AddRange(compressibleSensor.GetCompressedChannelMapping());
                 }
-                var dimensionPropertySensor = sensor as IDimensionPropertiesSensor;
-                if (dimensionPropertySensor != null)
+            }
+            // Add the dimension properties if any to the observationProto
+            var dimensionPropertySensor = sensor as IDimensionPropertiesSensor;
+            if (dimensionPropertySensor != null)
+            {
+                Debug.Log(shape[0] + " " + shape[1]);
+                var dimensionProperties = dimensionPropertySensor.GetDimensionProperties();
+                int[] intDimensionProperties = new int[dimensionProperties.Length];
+                for (int i = 0; i < dimensionProperties.Length; i++)
-                    var dimensionProperties = dimensionPropertySensor.GetDimensionProperties();
-                    int[] intDimensionProperties = new int[dimensionProperties.Length];
-                    for (int i = 0; i < dimensionProperties.Length; i++)
-                    {
-                        observationProto.DimensionProperties.Add((int)dimensionProperties[i]);
-                    }
+                    observationProto.DimensionProperties.Add((int)dimensionProperties[i]);
-
             }
             observationProto.Shape.AddRange(shape);
             return observationProto;

com.unity.ml-agents/Runtime/Sensors/BufferSensorComponent.cs

         /// <inheritdoc/>
         public override ISensor CreateSensor()
         {
-            m_Sensor = new BufferSensor(ObservableSize, MaxNumObservables);
+            m_Sensor = new BufferSensor(MaxNumObservables, ObservableSize);
             return m_Sensor;
         }