allow 'null' to be a value for job tier (Azure#34217)

sullnet · Feb 8, 2024 · 81d03b4 · 81d03b4
1 parent 3a024ef
commit 81d03b4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/constants/_job/job.py b/sdk/ml/azure-ai-ml/azure/ai/ml/constants/_job/job.py
@@ -119,6 +119,7 @@ class EntityNames:
         Basic = "basic"
         Standard = "standard"
         Premium = "premium"
+        Null = "Null"
 
     class RestNames:
         Spot = "Spot"
@@ -135,7 +136,7 @@ class RestNames:
 
     REST_TO_ENTITY = {v: k for k, v in ENTITY_TO_REST.items()}
 
-    ALLOWED_NAMES = [EntityNames.Spot, EntityNames.Basic, EntityNames.Standard, EntityNames.Premium]
+    ALLOWED_NAMES = [EntityNames.Spot, EntityNames.Basic, EntityNames.Standard, EntityNames.Premium, EntityNames.Null]
 
 
 class JobPriorityValues: