Fixed failing tests.

Unity-Technologies · Aug 25, 2024 · 0f801c7 · 0f801c7
1 parent 7f7ce0e
commit 0f801c7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/com.unity.ml-agents/Tests/Editor/Inference/EditModeTestInternalBrainTensorGenerator.cs b/com.unity.ml-agents/Tests/Editor/Inference/EditModeTestInternalBrainTensorGenerator.cs
@@ -171,10 +171,10 @@ public void GenerateActionMaskInput()
 
             generator.Generate(inputTensor, batchSize, inputs);
             Assert.IsNotNull(inputTensor.data);
-            Assert.AreEqual((int)((Tensor<int>)inputTensor.data)[0, 0], 1);
-            Assert.AreEqual((int)((Tensor<int>)inputTensor.data)[0, 4], 1);
-            Assert.AreEqual((int)((Tensor<int>)inputTensor.data)[1, 0], 0);
-            Assert.AreEqual((int)((Tensor<int>)inputTensor.data)[1, 4], 1);
+            Assert.AreEqual((int)((Tensor<float>)inputTensor.data)[0, 0], 1);
+            Assert.AreEqual((int)((Tensor<float>)inputTensor.data)[0, 4], 1);
+            Assert.AreEqual((int)((Tensor<float>)inputTensor.data)[1, 0], 0);
+            Assert.AreEqual((int)((Tensor<float>)inputTensor.data)[1, 4], 1);
         }
     }
 }
diff --git a/...ty.ml-agents/Tests/Editor/TestModels/discrete1vis0vec_2_3action_obsolete_recurr_v1_0.onnx b/...ty.ml-agents/Tests/Editor/TestModels/discrete1vis0vec_2_3action_obsolete_recurr_v1_0.onnx