Add api_version support to cluster start method (#256)

qubole · Jan 2, 2019 · c527090 · c527090
1 parent 925bacf
commit c527090
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qds_sdk/cluster.py b/qds_sdk/cluster.py
@@ -95,11 +95,11 @@ def status(cls, cluster_id_label):
         return conn.get(cls.element_path(cluster_id_label) + "/state")
 
     @classmethod
-    def start(cls, cluster_id_label):
+    def start(cls, cluster_id_label, api_version=None):
         """
         Start the cluster with id/label `cluster_id_label`.
         """
-        conn = Qubole.agent()
+        conn = Qubole.agent(version=api_version)
         data = {"state": "start"}
         return conn.put(cls.element_path(cluster_id_label) + "/state", data)