Add GPU Enabled flags

ajanikow · ajanikow · commit 4be36ba80b3d · 2024-04-25T16:42:35.000Z
diff --git a/docs/api/ArangoMLExtension.V1Alpha1.md b/docs/api/ArangoMLExtension.V1Alpha1.md
@@ -100,7 +100,7 @@ Links:
 
 Type: `boolean` <sup>[\[ref\]](https://github.com/arangodb/kube-arangodb/blob/1.2.40/pkg/apis/ml/v1alpha1/extension_spec_deployment.go#L52)</sup>
 
-GPU defined if GPU Jobs are enabledt.
+GPU defined if GPU Jobs are enabled.
 
 Default Value: `false`
 
diff --git a/pkg/apis/ml/v1alpha1/extension_spec_deployment.go b/pkg/apis/ml/v1alpha1/extension_spec_deployment.go
@@ -47,7 +47,7 @@ type ArangoMLExtensionSpecDeployment struct {
 	// Container Keeps the information about Container configuration
 	*schedulerContainerApi.Container `json:",inline"`
 
-	// GPU defined if GPU Jobs are enabledt.
+	// GPU defined if GPU Jobs are enabled.
 	// +doc/default: false
 	GPU *bool `json:"gpu,omitempty"`
 
diff --git a/pkg/apis/ml/v1alpha1/extension_spec_job.go b/pkg/apis/ml/v1alpha1/extension_spec_job.go
@@ -36,6 +36,14 @@ const (
 	MLJobFeaturizationType JobType = "featurization"
 )
 
+func MLJobTypes() []JobType {
+	return []JobType{
+		MLJobTrainingType,
+		MLJobPredictionType,
+		MLJobFeaturizationType,
+	}
+}
+
 type ArangoMLJobsTemplates struct {
 	// Prediction defines template for the prediction job
 	Prediction *ArangoMLJobTemplates `json:"prediction,omitempty"`
diff --git a/pkg/crd/crds/ml-extension.schema.generated.yaml b/pkg/crd/crds/ml-extension.schema.generated.yaml
@@ -419,7 +419,7 @@ v1alpha1:
                   type: object
                 type: array
               gpu:
-                description: GPU defined if GPU Jobs are enabledt.
+                description: GPU defined if GPU Jobs are enabled.
                 type: boolean
               hostIPC:
                 type: boolean
