add toolkit to sdk docs and example

Azure · Jun 11, 2018 · d688c9c · d688c9c
1 parent 2cb56f8
commit d688c9c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/docs/sdk-examples.md b/docs/sdk-examples.md
@@ -53,7 +53,8 @@ spark_conf = aztk.spark.models.SparkConfiguration(
 # configure my cluster
 cluster_config = aztk.spark.models.ClusterConfiguration(
     cluster_id="sdk-test",
-    vm_low_pri_count=2,
+    toolkit=aztk.spark.models.SparkToolkit(version="2.3.0"),
+    size_low_priority=2,
     vm_size="standard_f2",
     plugins=plugins,
     spark_configuration=spark_conf

diff --git a/examples/sdk/sdk_example.py b/examples/sdk/sdk_example.py
@@ -40,7 +40,8 @@
 # configure my cluster
 cluster_config = aztk.spark.models.ClusterConfiguration(
     cluster_id="sdk-test",
-    vm_low_pri_count=2,
+    toolkit=aztk.spark.models.SparkToolkit(version="2.3.0"),
+    size=2,
     vm_size="standard_f2",
     custom_scripts=[custom_script],
     spark_configuration=spark_conf