Skip to content

Commit 95ac6ae

Browse files
committed
Updated pr.
1 parent 98e11ff commit 95ac6ae

File tree

2 files changed

+7
-6
lines changed

2 files changed

+7
-6
lines changed

ads/model/deployment/model_deployment.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@
3636
from ads.model.deployment.model_deployment_runtime import (
3737
ModelDeploymentCondaRuntime,
3838
ModelDeploymentContainerRuntime,
39+
ModelDeploymentMode,
3940
ModelDeploymentRuntime,
4041
ModelDeploymentRuntimeType,
4142
OCIModelDeploymentRuntimeType,
@@ -85,11 +86,6 @@ class ModelDeploymentLogType:
8586
ACCESS = "access"
8687

8788

88-
class ModelDeploymentMode:
89-
HTTPS = "HTTPS_ONLY"
90-
STREAM = "STREAM_ONLY"
91-
92-
9389
class LogNotConfiguredError(Exception): # pragma: no cover
9490
pass
9591

ads/model/deployment/model_deployment_runtime.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,11 @@ class OCIModelDeploymentRuntimeType:
2121
CONTAINER = "OCIR_CONTAINER"
2222

2323

24+
class ModelDeploymentMode:
25+
HTTPS = "HTTPS_ONLY"
26+
STREAM = "STREAM_ONLY"
27+
28+
2429
class ModelDeploymentRuntime(Builder):
2530
"""A class used to represent a Model Deployment Runtime.
2631
@@ -173,7 +178,7 @@ def deployment_mode(self) -> str:
173178
str
174179
The deployment mode of model deployment.
175180
"""
176-
return self.get_spec(self.CONST_DEPLOYMENT_MODE, None)
181+
return self.get_spec(self.CONST_DEPLOYMENT_MODE, ModelDeploymentMode.HTTPS)
177182

178183
def with_deployment_mode(self, deployment_mode: str) -> "ModelDeploymentRuntime":
179184
"""Sets the deployment mode of model deployment.

0 commit comments

Comments
 (0)