Skip to content

Commit 150fe60

Browse files
vincentpierresurfnerd
authored andcommitted
Adding more tests for v2 models (#5138)
* Adding more tests for v2 models * formatting * fixing formatting
1 parent 9a4a4bf commit 150fe60

27 files changed

+89
-12
lines changed

com.unity.ml-agents/Tests/Editor/Analytics/InferenceAnalyticsTests.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ namespace Unity.MLAgents.Tests.Analytics
1515
[TestFixture]
1616
public class InferenceAnalyticsTests
1717
{
18-
const string k_continuousONNXPath = "Packages/com.unity.ml-agents/Tests/Editor/TestModels/continuous2vis8vec2action.onnx";
18+
const string k_continuousONNXPath = "Packages/com.unity.ml-agents/Tests/Editor/TestModels/continuous2vis8vec2action_v1_0.onnx";
1919
NNModel continuousONNXModel;
2020
Test3DSensorComponent sensor_21_20_3;
2121
Test3DSensorComponent sensor_20_22_3;

com.unity.ml-agents/Tests/Editor/BehaviorParameterTests.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ namespace Unity.MLAgents.Tests
1111
[TestFixture]
1212
public class BehaviorParameterTests : IHeuristicProvider
1313
{
14-
const string k_continuousONNXPath = "Packages/com.unity.ml-agents/Tests/Editor/TestModels/continuous2vis8vec2action.onnx";
14+
const string k_continuousONNXPath = "Packages/com.unity.ml-agents/Tests/Editor/TestModels/continuous2vis8vec2action_v1_0.onnx";
1515
public void Heuristic(in ActionBuffers actionsOut)
1616
{
1717
// No-op

com.unity.ml-agents/Tests/Editor/Inference.meta

Lines changed: 8 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.

com.unity.ml-agents/Tests/Editor/ModelRunnerTest.cs renamed to com.unity.ml-agents/Tests/Editor/Inference/ModelRunnerTest.cs

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,12 @@ namespace Unity.MLAgents.Tests
1212
[TestFixture]
1313
public class ModelRunnerTest
1414
{
15-
const string k_continuousONNXPath = "Packages/com.unity.ml-agents/Tests/Editor/TestModels/continuous2vis8vec2action.onnx";
16-
const string k_discreteONNXPath = "Packages/com.unity.ml-agents/Tests/Editor/TestModels/discrete1vis0vec_2_3action_recurr.onnx";
17-
const string k_hybridONNXPath = "Packages/com.unity.ml-agents/Tests/Editor/TestModels/hybrid0vis53vec_3c_2daction.onnx";
18-
const string k_continuousNNPath = "Packages/com.unity.ml-agents/Tests/Editor/TestModels/continuous2vis8vec2action_deprecated.nn";
19-
const string k_discreteNNPath = "Packages/com.unity.ml-agents/Tests/Editor/TestModels/discrete1vis0vec_2_3action_recurr_deprecated.nn";
15+
const string k_continuousONNXPath = "Packages/com.unity.ml-agents/Tests/Editor/TestModels/continuous2vis8vec2action_v1_0.onnx";
16+
const string k_discreteONNXPath = "Packages/com.unity.ml-agents/Tests/Editor/TestModels/discrete1vis0vec_2_3action_recurr_v1_0.onnx";
17+
const string k_hybridONNXPath = "Packages/com.unity.ml-agents/Tests/Editor/TestModels/hybrid0vis53vec_3c_2daction_v1_0.onnx";
18+
const string k_continuousNNPath = "Packages/com.unity.ml-agents/Tests/Editor/TestModels/continuous2vis8vec2action_deprecated_v1_0.nn";
19+
const string k_discreteNNPath = "Packages/com.unity.ml-agents/Tests/Editor/TestModels/discrete1vis0vec_2_3action_recurr_deprecated_v1_0.nn";
20+
2021
NNModel continuousONNXModel;
2122
NNModel discreteONNXModel;
2223
NNModel hybridONNXModel;
@@ -25,6 +26,7 @@ public class ModelRunnerTest
2526
Test3DSensorComponent sensor_21_20_3;
2627
Test3DSensorComponent sensor_20_22_3;
2728

29+
2830
ActionSpec GetContinuous2vis8vec2actionActionSpec()
2931
{
3032
return ActionSpec.MakeContinuous(2);

0 commit comments

Comments
 (0)