diff --git a/com.unity.ml-agents/Runtime/Analytics/Events.cs b/com.unity.ml-agents/Runtime/Analytics/Events.cs index fe0552d962..f269d91436 100644 --- a/com.unity.ml-agents/Runtime/Analytics/Events.cs +++ b/com.unity.ml-agents/Runtime/Analytics/Events.cs @@ -97,6 +97,7 @@ internal struct EventObservationSpec public string SensorName; public string CompressionType; public int BuiltInSensorType; + public int ObservationType; public EventObservationDimensionInfo[] DimensionInfos; public static EventObservationSpec FromSensor(ISensor sensor) @@ -119,6 +120,7 @@ public static EventObservationSpec FromSensor(ISensor sensor) SensorName = sensor.GetName(), CompressionType = sensor.GetCompressionSpec().SensorCompressionType.ToString(), BuiltInSensorType = (int)builtInSensorType, + ObservationType = (int)obsSpec.ObservationType, DimensionInfos = dimInfos, }; } diff --git a/com.unity.ml-agents/Tests/Editor/Analytics/InferenceAnalyticsTests.cs b/com.unity.ml-agents/Tests/Editor/Analytics/InferenceAnalyticsTests.cs index 79fcad81f7..7686517b6b 100644 --- a/com.unity.ml-agents/Tests/Editor/Analytics/InferenceAnalyticsTests.cs +++ b/com.unity.ml-agents/Tests/Editor/Analytics/InferenceAnalyticsTests.cs @@ -65,6 +65,7 @@ public void TestModelEvent() Assert.AreEqual(2, continuousEvent.ObservationSpecs.Count); Assert.AreEqual(3, continuousEvent.ObservationSpecs[0].DimensionInfos.Length); Assert.AreEqual(20, continuousEvent.ObservationSpecs[0].DimensionInfos[0].Size); + Assert.AreEqual(0, continuousEvent.ObservationSpecs[0].ObservationType); Assert.AreEqual((int)DimensionProperty.TranslationalEquivariance, continuousEvent.ObservationSpecs[0].DimensionInfos[0].Flags); Assert.AreEqual((int)DimensionProperty.None, continuousEvent.ObservationSpecs[0].DimensionInfos[2].Flags); Assert.AreEqual("None", continuousEvent.ObservationSpecs[0].CompressionType); diff --git a/com.unity.ml-agents/Tests/Editor/Analytics/TrainingAnalyticsTest.cs b/com.unity.ml-agents/Tests/Editor/Analytics/TrainingAnalyticsTest.cs index 4bfa8196ac..1010b4549b 100644 --- a/com.unity.ml-agents/Tests/Editor/Analytics/TrainingAnalyticsTest.cs +++ b/com.unity.ml-agents/Tests/Editor/Analytics/TrainingAnalyticsTest.cs @@ -41,6 +41,7 @@ public void TestRemotePolicyEvent() Assert.AreEqual(2, remotePolicyEvent.ObservationSpecs.Count); Assert.AreEqual(3, remotePolicyEvent.ObservationSpecs[0].DimensionInfos.Length); Assert.AreEqual(20, remotePolicyEvent.ObservationSpecs[0].DimensionInfos[0].Size); + Assert.AreEqual(0, remotePolicyEvent.ObservationSpecs[0].ObservationType); Assert.AreEqual("None", remotePolicyEvent.ObservationSpecs[0].CompressionType); Assert.AreEqual(Test3DSensor.k_BuiltInSensorType, remotePolicyEvent.ObservationSpecs[0].BuiltInSensorType);