[bug fix] Fix warning using demo recorder (#5216)

vincentpierre · web-flow · commit 272899a8a100 · 2021-04-01T13:31:55.000-07:00
diff --git a/com.unity.ml-agents/Runtime/Agent.cs b/com.unity.ml-agents/Runtime/Agent.cs
@@ -572,14 +572,15 @@ void NotifyAgentDone(DoneReason doneReason)
             // Request the last decision with no callbacks
             // We request a decision so Python knows the Agent is done immediately
             m_Brain?.RequestDecision(m_Info, sensors);
-            ResetSensors();
 
             // We also have to write any to any DemonstationStores so that they get the "done" flag.
             foreach (var demoWriter in DemonstrationWriters)
             {
                 demoWriter.Record(m_Info, sensors);
             }
 
+            ResetSensors();
+
             if (doneReason != DoneReason.Disabled)
             {
                 // We don't want to update the reward stats when the Agent is disabled, because this will make

Original file line number	Diff line number	Diff line change
`@@ -572,14 +572,15 @@ void NotifyAgentDone(DoneReason doneReason)`
`572`	`572`	`// Request the last decision with no callbacks`
`573`	`573`	`// We request a decision so Python knows the Agent is done immediately`
`574`	`574`	`m_Brain?.RequestDecision(m_Info, sensors);`
`575`		`- ResetSensors();`
`576`	`575`
`577`	`576`	`// We also have to write any to any DemonstationStores so that they get the "done" flag.`
`578`	`577`	`foreach (var demoWriter in DemonstrationWriters)`
`579`	`578`	`{`
`580`	`579`	`demoWriter.Record(m_Info, sensors);`
`581`	`580`	`}`
`582`	`581`
	`582`	`+ ResetSensors();`
	`583`	`+`
`583`	`584`	`if (doneReason != DoneReason.Disabled)`
`584`	`585`	`{`
`585`	`586`	`// We don't want to update the reward stats when the Agent is disabled, because this will make`