From c01a74c8272dc7ab03ac9fbd4018a1e427c6c0da Mon Sep 17 00:00:00 2001
From: Nicola Gatto <nicola.gatto@rwth-aachen.de>
Date: Sat, 8 Jun 2019 19:06:45 +0200
Subject: [PATCH] Always return zero if sample games is zero

---
 .../resources/templates/gluon/reinforcement/agent/Agent.ftl    | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/main/resources/templates/gluon/reinforcement/agent/Agent.ftl b/src/main/resources/templates/gluon/reinforcement/agent/Agent.ftl
index b37c8e29..5525f733 100644
--- a/src/main/resources/templates/gluon/reinforcement/agent/Agent.ftl
+++ b/src/main/resources/templates/gluon/reinforcement/agent/Agent.ftl
@@ -177,6 +177,9 @@ class Agent(object):
         return states, actions, rewards, next_states, terminals
 
     def evaluate(self, target=None, sample_games=100, verbose=True):
+        if sample_games <= 0:
+            return 0
+
         target = self._target_score if target is None else target
         if target:
             target_achieved = 0
-- 
GitLab