From 0f801c744dd7ed8e2a6f1f0a020da06bc6303207 Mon Sep 17 00:00:00 2001
From: Miguel Alonso Jr <miguel.alonso@unity3d.com>
Date: Sun, 25 Aug 2024 08:29:08 -0400
Subject: [PATCH] Fixed failing tests.

---
 ...ditModeTestInternalBrainTensorGenerator.cs |   8 ++++----
 ...is0vec_2_3action_obsolete_recurr_v1_0.onnx | Bin 399678 -> 399647 bytes
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/com.unity.ml-agents/Tests/Editor/Inference/EditModeTestInternalBrainTensorGenerator.cs b/com.unity.ml-agents/Tests/Editor/Inference/EditModeTestInternalBrainTensorGenerator.cs
index 43e21996ea..d09e0884af 100644
--- 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/com.unity.ml-agents/Tests/Editor/TestModels/discrete1vis0vec_2_3action_obsolete_recurr_v1_0.onnx b/com.unity.ml-agents/Tests/Editor/TestModels/discrete1vis0vec_2_3action_obsolete_recurr_v1_0.onnx
index e56d6c44dfcb4257afe27f1fc7f3754502f45d39..590906bc674794f452da209d4eb36a35c88d64fa 100644
GIT binary patch
delta 52
zcmdnDNn-vc2_X(PA@+jGlKi6N3@K(qJu|EJMxj<A##SMwRw3qAA(mDl)~!Npb0nsp
Im0&9b0GEOeumAu6

delta 67
zcmbQgNn+n72_X(PA@+jGlKi6N3@K(qJu|CijY6$LjIBaUtwPMLLM*L9tXqZH=16eL
Xa<L^Qmt^MWO@AQHCOUns1X~dRkB}1$