diff --git a/pyproject.toml b/pyproject.toml index 75d3b2b..0f163f4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "ds-platform-utils" -version = "0.2.2" +version = "0.2.3" description = "Utility library for Pattern Data Science." readme = "README.md" authors = [ diff --git a/src/ds_platform_utils/metaflow/write_audit_publish.py b/src/ds_platform_utils/metaflow/write_audit_publish.py index c54a74f..04423a6 100644 --- a/src/ds_platform_utils/metaflow/write_audit_publish.py +++ b/src/ds_platform_utils/metaflow/write_audit_publish.py @@ -1,4 +1,5 @@ import json +import os import warnings from pathlib import Path from textwrap import dedent @@ -119,7 +120,10 @@ def extract(prefix: str, default: str = "unknown") -> str: "user": current.username, # username of user who triggered the run (argo-workflows if its a deployed flow) "domain": extract("ds.domain"), # business unit (domain) of the flow, same as app "namespace": current.namespace, # namespace of the flow - "perimeter": "PROD" if current.is_production else "Default", # perimeter of the flow + "perimeter": str(os.environ.get("OB_CURRENT_PERIMETER") or os.environ.get("OBP_PERIMETER")), + "is_production": str( + current.is_production + ), # True, if the flow is deployed with the --production flag else false "team": "data-science", # team name, hardcoded as data-science } diff --git a/uv.lock b/uv.lock index f2a943c..de6dc7f 100644 --- a/uv.lock +++ b/uv.lock @@ -478,7 +478,7 @@ wheels = [ [[package]] name = "ds-platform-utils" -version = "0.2.2" +version = "0.2.3" source = { editable = "." } dependencies = [ { name = "jinja2" },