Removing warning in the Agent Inspector when Vis Obs is used

vincentpierre · vincentpierre · commit c0014e6f9683 · 2019-10-22T16:36:46.000-07:00
diff --git a/UnitySDK/Assets/ML-Agents/Scripts/InferenceBrain/BarracudaModelParamLoader.cs b/UnitySDK/Assets/ML-Agents/Scripts/InferenceBrain/BarracudaModelParamLoader.cs
@@ -338,8 +338,11 @@ private static IEnumerable<string> CheckInputTensorShape(
             {
                 if (!tensorTester.ContainsKey(tensor.name))
                 {
-                    failedModelChecks.Add(
-                        "Model requires an unknown input named : " + tensor.name);
+                    if (!tensor.name.Contains("visual_observation"))
+                    {
+                        failedModelChecks.Add(
+                            "Model requires an unknown input named : " + tensor.name);
+                    }
                 }
                 else
                 {

Original file line number	Diff line number	Diff line change
`@@ -338,8 +338,11 @@ private static IEnumerable<string> CheckInputTensorShape(`
`338`	`338`	`{`
`339`	`339`	`if (!tensorTester.ContainsKey(tensor.name))`
`340`	`340`	`{`
`341`		`- failedModelChecks.Add(`
`342`		`- "Model requires an unknown input named : " + tensor.name);`
	`341`	`+ if (!tensor.name.Contains("visual_observation"))`
	`342`	`+ {`
	`343`	`+ failedModelChecks.Add(`
	`344`	`+ "Model requires an unknown input named : " + tensor.name);`
	`345`	`+ }`
`343`	`346`	`}`
`344`	`347`	`else`
`345`	`348`	`{`