File tree Expand file tree Collapse file tree 2 files changed +7
-6
lines changed Expand file tree Collapse file tree 2 files changed +7
-6
lines changed Original file line number Diff line number Diff line change 36
36
from ads .model .deployment .model_deployment_runtime import (
37
37
ModelDeploymentCondaRuntime ,
38
38
ModelDeploymentContainerRuntime ,
39
+ ModelDeploymentMode ,
39
40
ModelDeploymentRuntime ,
40
41
ModelDeploymentRuntimeType ,
41
42
OCIModelDeploymentRuntimeType ,
@@ -85,11 +86,6 @@ class ModelDeploymentLogType:
85
86
ACCESS = "access"
86
87
87
88
88
- class ModelDeploymentMode :
89
- HTTPS = "HTTPS_ONLY"
90
- STREAM = "STREAM_ONLY"
91
-
92
-
93
89
class LogNotConfiguredError (Exception ): # pragma: no cover
94
90
pass
95
91
Original file line number Diff line number Diff line change @@ -21,6 +21,11 @@ class OCIModelDeploymentRuntimeType:
21
21
CONTAINER = "OCIR_CONTAINER"
22
22
23
23
24
+ class ModelDeploymentMode :
25
+ HTTPS = "HTTPS_ONLY"
26
+ STREAM = "STREAM_ONLY"
27
+
28
+
24
29
class ModelDeploymentRuntime (Builder ):
25
30
"""A class used to represent a Model Deployment Runtime.
26
31
@@ -173,7 +178,7 @@ def deployment_mode(self) -> str:
173
178
str
174
179
The deployment mode of model deployment.
175
180
"""
176
- return self .get_spec (self .CONST_DEPLOYMENT_MODE , None )
181
+ return self .get_spec (self .CONST_DEPLOYMENT_MODE , ModelDeploymentMode . HTTPS )
177
182
178
183
def with_deployment_mode (self , deployment_mode : str ) -> "ModelDeploymentRuntime" :
179
184
"""Sets the deployment mode of model deployment.
You can’t perform that action at this time.
0 commit comments