Skip to content

Commit 764d8fc

Browse files
committed
Added compartment id environment variable while running in pipeline run.
1 parent a763f81 commit 764d8fc

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

ads/model/generic_model.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
JOB_RUN_OCID,
3030
NB_SESSION_COMPARTMENT_OCID,
3131
NB_SESSION_OCID,
32+
PIPELINE_RUN_COMPARTMENT_OCID,
3233
PROJECT_OCID,
3334
)
3435
from ads.evaluations import EvaluatorMixin
@@ -91,7 +92,11 @@
9192
from ads.model.transformer.onnx_transformer import ONNXTransformer
9293

9394
_TRAINING_RESOURCE_ID = JOB_RUN_OCID or NB_SESSION_OCID
94-
_COMPARTMENT_OCID = NB_SESSION_COMPARTMENT_OCID or JOB_RUN_COMPARTMENT_OCID
95+
_COMPARTMENT_OCID = (
96+
NB_SESSION_COMPARTMENT_OCID
97+
or JOB_RUN_COMPARTMENT_OCID
98+
or PIPELINE_RUN_COMPARTMENT_OCID
99+
)
95100

96101
MODEL_DEPLOYMENT_INSTANCE_SHAPE = "VM.Standard.E4.Flex"
97102
MODEL_DEPLOYMENT_INSTANCE_OCPUS = 1

0 commit comments

Comments
 (0)