Add support for PythonWheelTask (#872)

* Add support for PythonWheelTask

Co-authored-by: Serge Smertin <[email protected]>

Author: tingting-gao-db

compute/model.go

@@ -517,6 +517,14 @@ type SparkSubmitTask struct {
 	Parameters []string `json:"parameters,omitempty"`
 }
 
+// PythonWheelTask contains the information for python wheel jobs
+type PythonWheelTask struct {
+    EntryPoint      string            `json:"entry_point,omitempty"`
+    PackageName     string            `json:"package_name,omitempty"`
+    Parameters      []string          `json:"parameters,omitempty"`
+    NamedParameters map[string]string `json:"named_parameters,omitempty"`
+}
+
 // PipelineTask contains the information for pipeline jobs
 type PipelineTask struct {
 	PipelineID string `json:"pipeline_id"`
@@ -546,7 +554,6 @@ type JobTaskSettings struct {
 	Description string           `json:"description,omitempty"`
 	DependsOn   []TaskDependency `json:"depends_on,omitempty"`
 
-	// TODO: add PythonWheelTask
 	ExistingClusterID      string              `json:"existing_cluster_id,omitempty" tf:"group:cluster_type"`
 	NewCluster             *Cluster            `json:"new_cluster,omitempty" tf:"group:cluster_type"`
 	Libraries              []Library           `json:"libraries,omitempty" tf:"slice_set,alias:library"`
@@ -555,6 +562,7 @@ type JobTaskSettings struct {
 	SparkPythonTask        *SparkPythonTask    `json:"spark_python_task,omitempty" tf:"group:task_type"`
 	SparkSubmitTask        *SparkSubmitTask    `json:"spark_submit_task,omitempty" tf:"group:task_type"`
 	PipelineTask           *PipelineTask       `json:"pipeline_task,omitempty" tf:"group:task_type"`
+  PythonWheelTask        *PythonWheelTask    `json:"python_wheel_task,omitempty" tf:"group:task_type"`
 	EmailNotifications     *EmailNotifications `json:"email_notifications,omitempty" tf:"suppress_diff"`
 	TimeoutSeconds         int32               `json:"timeout_seconds,omitempty"`
 	MaxRetries             int32               `json:"max_retries,omitempty"`
@@ -574,7 +582,8 @@ type JobSettings struct {
 	SparkPythonTask        *SparkPythonTask `json:"spark_python_task,omitempty" tf:"group:task_type"`
 	SparkSubmitTask        *SparkSubmitTask `json:"spark_submit_task,omitempty" tf:"group:task_type"`
 	PipelineTask           *PipelineTask    `json:"pipeline_task,omitempty" tf:"group:task_type"`
-	Libraries              []Library        `json:"libraries,omitempty" tf:"slice_set,alias:library"`
+	PythonWheelTask        *PythonWheelTask `json:"python_wheel_task,omitempty" tf:"group:task_type"`
+  Libraries              []Library        `json:"libraries,omitempty" tf:"slice_set,alias:library"`
 	TimeoutSeconds         int32            `json:"timeout_seconds,omitempty"`
 	MaxRetries             int32            `json:"max_retries,omitempty"`
 	MinRetryIntervalMillis int32            `json:"min_retry_interval_millis,omitempty"`