From 2b4c9c67748cf1447be7fb5053748806977d67ce Mon Sep 17 00:00:00 2001
From: Gal Leibovich <gal.leibovich@intel.com>
Date: Fri, 23 Nov 2018 21:42:54 +0200
Subject: [PATCH] Removing grarph_manager param (#141)

---
 rl_coach/coach.py                  | 2 +-
 rl_coach/memories/backend/redis.py | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/rl_coach/coach.py b/rl_coach/coach.py
index 2542eb4..7c0274f 100644
--- a/rl_coach/coach.py
+++ b/rl_coach/coach.py
@@ -579,7 +579,7 @@ class CoachLauncher(object):
             return
 
         if args.distributed_coach and args.distributed_coach_run_type == RunType.ORCHESTRATOR:
-            handle_distributed_coach_orchestrator(graph_manager, args)
+            handle_distributed_coach_orchestrator(args)
             return
 
         # Single-threaded runs
diff --git a/rl_coach/memories/backend/redis.py b/rl_coach/memories/backend/redis.py
index 9a94880..b6010a0 100644
--- a/rl_coach/memories/backend/redis.py
+++ b/rl_coach/memories/backend/redis.py
@@ -113,6 +113,8 @@ class RedisPubSubBackend(MemoryBackend):
     def undeploy(self):
         if self.params.deployed:
             return
+
+        from kubernetes import client
         api_client = client.AppsV1Api()
         delete_options = client.V1DeleteOptions()
         try: