Merge pull request #115 from mrocklin/return-close

Pass through return value from close
dask · Dec 22, 2018 · 7a78a0d · 7a78a0d
2 parents 25fb05f + c92414b
commit 7a78a0d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dask_kubernetes/core.py b/dask_kubernetes/core.py
@@ -485,10 +485,10 @@ def scale_down(self, workers, pods=None):
     def __enter__(self):
         return self
 
-    def close(self):
+    def close(self, **kwargs):
         """ Close this cluster """
         self.scale_down(self.cluster.scheduler.workers)
-        self.cluster.close()
+        return self.cluster.close(**kwargs)
 
     def __exit__(self, type, value, traceback):
         _cleanup_pods(self.namespace, self.pod_template.metadata.labels)