Fixes AWS api being called when sim job is cancelled

aws-deepracer-community · Jun 30, 2019 · 3df4b46 · 3df4b46
1 parent abb2c95
commit 3df4b46
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/rl_coach/src/markov/environments/deepracer_racetrack_env.py b/rl_coach/src/markov/environments/deepracer_racetrack_env.py
@@ -511,11 +511,13 @@ def is_training_done(self):
         return self.is_simulation_done
 
     def cancel_simulation_job(self):
-        session = boto3.session.Session()
-        robomaker_client = session.client('robomaker', region_name=self.aws_region)
-        robomaker_client.cancel_simulation_job(
-            job=self.simulation_job_arn
-        )
+        isLocal = os.environ.get("LOCAL")
+        if isLocal != None:
+            session = boto3.session.Session()
+            robomaker_client = session.client('robomaker', region_name=self.aws_region)
+            robomaker_client.cancel_simulation_job(
+                job=self.simulation_job_arn
+            )
 
     def send_reward_to_cloudwatch(self, reward):
         isLocal = os.environ.get("LOCAL")