Use GoalSensor in GridWorld

Project/Assets/ML-Agents/Examples/GridWorld/Prefabs/Area.prefab

   - component: {fileID: 114650561397225712}
   - component: {fileID: 114889700908650620}
   - component: {fileID: 7980686505185502968}
+  - component: {fileID: 8359584214300847863}
   m_Layer: 8
   m_Name: Agent
   m_TagString: agent
   m_Name: 
   m_EditorClassIdentifier: 
   m_BrainParameters:
-    VectorObservationSize: 2
+    VectorObservationSize: 0
     NumStackedVectorObservations: 1
     VectorActionSize: 05000000
     VectorActionDescriptions: []
   m_Name: 
   m_EditorClassIdentifier: 
   debugCommandLineOverride: 
+--- !u!114 &8359584214300847863
+MonoBehaviour:
+  m_ObjectHideFlags: 0
+  m_CorrespondingSourceObject: {fileID: 0}
+  m_PrefabInstance: {fileID: 0}
+  m_PrefabAsset: {fileID: 0}
+  m_GameObject: {fileID: 1488387672112076}
+  m_Enabled: 1
+  m_EditorHideFlags: 0
+  m_Script: {fileID: 11500000, guid: 163dac4bcbb2f4d8499db2cdcb22a89e, type: 3}
+  m_Name: 
+  m_EditorClassIdentifier: 
+  observationSize: 1
 --- !u!1 &1625008366184734
 GameObject:
   m_ObjectHideFlags: 0

Project/Assets/ML-Agents/Examples/GridWorld/Scenes/GridWorld.unity

   - component: {fileID: 125487790}
   - component: {fileID: 125487787}
   - component: {fileID: 125487791}
+  - component: {fileID: 125487792}
   m_Layer: 8
   m_Name: RenderTextureAgent
   m_TagString: agent
   m_Name: 
   m_EditorClassIdentifier: 
   m_BrainParameters:
-    VectorObservationSize: 2
+    VectorObservationSize: 0
     NumStackedVectorObservations: 1
     VectorActionSize: 05000000
     VectorActionDescriptions: []
   m_Grayscale: 0
   m_ObservationStacks: 1
   m_Compression: 1
+--- !u!114 &125487792
+MonoBehaviour:
+  m_ObjectHideFlags: 0
+  m_CorrespondingSourceObject: {fileID: 0}
+  m_PrefabInstance: {fileID: 0}
+  m_PrefabAsset: {fileID: 0}
+  m_GameObject: {fileID: 125487785}
+  m_Enabled: 1
+  m_EditorHideFlags: 0
+  m_Script: {fileID: 11500000, guid: 163dac4bcbb2f4d8499db2cdcb22a89e, type: 3}
+  m_Name: 
+  m_EditorClassIdentifier: 
+  observationSize: 1
 --- !u!1 &260425459
 GameObject:
   m_ObjectHideFlags: 0
     - target: {fileID: 114935253044749092, guid: 5c2bd19e4bbda4991b74387ca5d28156,
         type: 3}
       propertyPath: m_BrainParameters.VectorObservationSize
-      value: 2
+      value: 0
       objectReference: {fileID: 0}
     - target: {fileID: 114935253044749092, guid: 5c2bd19e4bbda4991b74387ca5d28156,
         type: 3}

Project/Assets/ML-Agents/Examples/GridWorld/Scripts/GridAgent.cs

         "masking turned on may not behave optimally when action masking is turned off.")]
     public bool maskActions = true;
 
+    GoalSensorComponent goalSensor;
 
     public GridGoal gridGoal;
 
     {
         Array values = Enum.GetValues(typeof(GridGoal));
         int goalNum = (int)gridGoal;
-        sensor.AddOneHotObservation(goalNum, values.Length);
+        goalSensor = this.GetComponent<GoalSensorComponent>();
+        goalSensor.AddGoal(goalNum);
     }
 
     public override void WriteDiscreteActionMask(IDiscreteActionMask actionMask)

ml-agents-envs/mlagents_envs/base_env.py

 
 class SensorType(Enum):
     OBSERVATION = 0
-    PARAMETERIZATION = 1
+    GOAL = 1
+
 
 class _ActionTupleBase(ABC):
     """