Fix test reward function to new CoCo check

9ea86925 · Nicola Gatto · 3c4ca5bc · 9ea86925
Commit 9ea86925 authored May 06, 2019 by Nicola Gatto
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 2 deletions

src/test/resources/models/reinforcementModel/cartpole/agent/reward/Reward.emadl ...els/reinforcementModel/cartpole/agent/reward/Reward.emadl +6 -2

No files found.
--- a/src/test/resources/models/reinforcementModel/cartpole/agent/reward/Reward.emadl
+++ b/src/test/resources/models/reinforcementModel/cartpole/agent/reward/Reward.emadl
@@ -3,10 +3,14 @@ package cartpole.agent.reward;
 component Reward {
    ports
        in Q^{4} state,
+        in B isTerminal,
        out Q reward;

    implementation Math {
-        Q rew = state(1);
-        reward = rew;
+        if isTerminal
+            reward = -1;
+        else
+            reward = 1;
+        end
    }
 }
\ No newline at end of file