polyaxon 2.0.0rc49__py3-none-any.whl → 2.4.0rc1__py3-none-any.whl

polyaxon/_client/run.py

@@ -21,6 +21,7 @@ from clipped.utils.paths import (
     get_base_filename,
     get_dirs_under_path,
     get_files_in_path_context,
+    set_permissions,
 )
 from clipped.utils.query_params import (
     get_logs_params,
@@ -35,6 +36,7 @@ from polyaxon import settings
 from polyaxon._cli.errors import handle_cli_error
 from polyaxon._client.client import PolyaxonClient
 from polyaxon._client.decorators import client_handler, get_global_or_inline_config
+from polyaxon._client.mixin import ClientMixin
 from polyaxon._client.store import PolyaxonStore
 from polyaxon._constants.metadata import META_COPY_ARTIFACTS, META_RECOMPILE
 from polyaxon._containers.names import MAIN_CONTAINER_NAMES
@@ -58,11 +60,17 @@ from polyaxon._schemas.lifecycle import (
     V1Statuses,
 )
 from polyaxon._schemas.types import V1ArtifactsType
+from polyaxon._sdk.schemas import V1RunEdgeLineage, V1RunEdgesGraph
 from polyaxon._sdk.schemas.v1_operation_body import V1OperationBody
 from polyaxon._sdk.schemas.v1_project_version import V1ProjectVersion
 from polyaxon._sdk.schemas.v1_run import V1Run
 from polyaxon._sdk.schemas.v1_run_settings import V1RunSettings
-from polyaxon._utils.fqn_utils import get_entity_full_name, to_fqn_name
+from polyaxon._sdk.schemas.v1_settings_catalog import V1SettingsCatalog
+from polyaxon._utils.fqn_utils import (
+    get_entity_full_name,
+    split_owner_team_space,
+    to_fqn_name,
+)
 from polyaxon._utils.urls_utils import get_proxy_run_url
 from polyaxon.api import K8S_V1_LOCATION, STREAMS_V1_LOCATION
 from polyaxon.exceptions import ApiException, PolyaxonClientException
@@ -79,7 +87,7 @@ if TYPE_CHECKING:
     from traceml.tracking.run import Run
 
 
-class RunClient:
+class RunClient(ClientMixin):
     """RunClient is a client to communicate with Polyaxon runs endpoints.
 
     If no values are passed to this class,
@@ -145,7 +153,7 @@ class RunClient:
             return
 
         try:
-            owner, project = get_project_or_local(
+            owner, _, project = get_project_or_local(
                 get_entity_full_name(owner=owner, entity=project)
             )
         except PolyaxonClientException:
@@ -165,8 +173,10 @@ class RunClient:
         if error_message and not self._is_offline:
             raise PolyaxonClientException(error_message)
 
+        owner, team = split_owner_team_space(owner)
         self._client = client
         self._owner = owner
+        self._team = team
         self._project = project
         self._run_uuid = (
             get_run_or_local(run_uuid)
@@ -217,12 +227,23 @@ class RunClient:
             return client.config.no_op
         return settings.CLIENT_CONFIG.no_op
 
-    @property
-    def client(self):
-        if self._client:
-            return self._client
-        self._client = PolyaxonClient()
-        return self._client
+    def _reset_agent(self, agent: Optional[str] = None):
+        if agent:
+            agent = self.client.agents_v1.get_agent(self.owner, agent)
+            self.settings.agent = V1SettingsCatalog(
+                uuid=agent.uuid,
+                name=agent.name,
+                version=agent.version,
+                url=agent.hostname,
+            )
+            self.settings.namespace = agent.namespace
+            self.settings.artifacts_store = None
+
+    def _use_agent_host(self):
+        if self.settings.agent and self.settings.agent.url:
+            self.reset_client(
+                host=self.settings.agent.url, POLYAXON_HOST=self.settings.agent.url
+            )
 
     @property
     def store(self):
@@ -255,20 +276,6 @@ class RunClient:
             return self.settings.artifacts_store.name
         return None
 
-    @property
-    def owner(self) -> str:
-        return self._owner or ""
-
-    def set_owner(self, owner: str):
-        self._owner = owner
-
-    @property
-    def project(self) -> str:
-        return self._project or ""
-
-    def set_project(self, project: str):
-        self._project = project
-
     @property
     def run_uuid(self) -> str:
         return self._run_uuid
@@ -322,7 +329,7 @@ class RunClient:
             lineages = self.get_artifacts_lineage(limit=1000).results
             self._artifacts_lineage = {l.name: l for l in lineages}
         if load_metrics:
-            self.get_metrics(self.metric_names, force)
+            self.get_metrics(self._metric_names, force=True)
 
     def _throttle_updates(self) -> bool:
         current_time = now().replace(microsecond=0)
@@ -500,6 +507,7 @@ class RunClient:
         matrix: Optional[Union[Dict, V1Matrix]] = None,
         presets: Optional[List[str]] = None,
         queue: Optional[str] = None,
+        namespace: Optional[str] = None,
         nocache: Optional[bool] = None,
         cache: Optional[Union[int, str, bool]] = None,
         approved: Optional[Union[int, str, bool]] = None,
@@ -529,6 +537,8 @@ class RunClient:
                  [presets](/docs/core/scheduling-presets/).
             queue: str, optional, the name of the
                  [queue](/docs/core/scheduling-strategies/queues/) to assign the run to.
+            namespace: str, optional, the namespace to create the run under, by default it will
+                 use the agent's namespace.
             nocache: bool, optional, DEPRECATED Please use `cache='f'`
                  simple flag to disable
                  [cache check](/docs/automation/helpers/cache/).
@@ -554,6 +564,7 @@ class RunClient:
             matrix=matrix,
             presets=presets,
             queue=queue,
+            namespace=namespace,
             nocache=nocache,
             cache=cache,
             approved=approved,
@@ -575,6 +586,7 @@ class RunClient:
         matrix: Optional[Union[Dict, V1Matrix]] = None,
         presets: Optional[List[str]] = None,
         queue: Optional[str] = None,
+        namespace: Optional[str] = None,
         nocache: Optional[bool] = None,
         cache: Optional[Union[int, str, bool]] = None,
         approved: Optional[Union[int, str, bool]] = None,
@@ -603,6 +615,8 @@ class RunClient:
                  [presets](/docs/core/scheduling-presets/).
             queue: str, optional, the name of the
                  [queue](/docs/core/scheduling-strategies/queues/) to assign the run to.
+            namespace: str, optional, the namespace to create the run under, by default it will
+                 use the agent's namespace.
             nocache: bool, optional, DEPRECATED Please use `cache='f'`
                  simple flag to disable
                  [cache check](/docs/automation/helpers/cache/).
@@ -628,6 +642,7 @@ class RunClient:
             matrix=matrix,
             presets=presets,
             queue=queue,
+            namespace=namespace,
             nocache=nocache,
             cache=cache,
             approved=approved,
@@ -649,6 +664,7 @@ class RunClient:
         matrix: Optional[Union[Dict, V1Matrix]] = None,
         presets: Optional[List[str]] = None,
         queue: Optional[str] = None,
+        namespace: Optional[str] = None,
         nocache: Optional[bool] = None,
         cache: Optional[Union[int, str, bool]] = None,
         approved: Optional[Union[int, str, bool]] = None,
@@ -676,6 +692,8 @@ class RunClient:
                  [presets](/docs/core/scheduling-presets/).
             queue: str, optional, the name of the
                  [queue](/docs/core/scheduling-strategies/queues/) to assign the run to.
+            namespace: str, optional, the namespace to create the run under, by default it will
+                 use the agent's namespace.
             nocache: bool, optional, DEPRECATED Please use `cache='f'`
                  simple flag to disable
                  [cache check](/docs/automation/helpers/cache/).
@@ -701,6 +719,7 @@ class RunClient:
             matrix=matrix,
             presets=presets,
             queue=queue,
+            namespace=namespace,
             nocache=nocache,
             cache=cache,
             approved=approved,
@@ -885,6 +904,7 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
         params = get_logs_params(
             last_file=last_file, last_time=last_time, connection=self.artifacts_store
         )
@@ -908,6 +928,7 @@ class RunClient:
     def inspect(self):
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
         params = get_streams_params(connection=self.artifacts_store, status=self.status)
         return self.client.runs_v1.inspect_run(
             self.namespace, self.owner, self.project, self.run_uuid, **params
@@ -981,6 +1002,10 @@ class RunClient:
                 if not container:
                     container = pod_containers[0]
 
+        if not self.settings:
+            self.refresh_data()
+        self._use_agent_host()
+
         url = get_proxy_run_url(
             service=K8S_V1_LOCATION,
             namespace=self.namespace,
@@ -1026,6 +1051,8 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
+
         params = get_streams_params(self.artifacts_store)
         return self.client.runs_v1.get_run_events(
             self.namespace,
@@ -1061,14 +1088,16 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
+
         params = get_streams_params(self.artifacts_store)
         return self.client.runs_v1.get_multi_run_events(
             self.namespace,
             self.owner,
             self.project,
             kind=kind,
-            names=names,
-            runs=runs,
+            names=",".join(names),
+            runs=",".join(runs),
             orient=orient,
             force=force,
             **params,
@@ -1297,6 +1326,7 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
         params = get_streams_params(self.artifacts_store)
         return self.client.runs_v1.get_run_artifact(
             namespace=self.namespace,
@@ -1330,6 +1360,10 @@ class RunClient:
         if not self.run_uuid:
             return
 
+        if not self.settings:
+            self.refresh_data()
+        self._use_agent_host()
+
         lineage_path = lineage.path or ""
         summary = lineage.summary or {}
         is_event = summary.get("is_event")
@@ -1408,6 +1442,8 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
+
         url = get_proxy_run_url(
             service=STREAMS_V1_LOCATION,
             namespace=self.namespace,
@@ -1447,6 +1483,8 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
+
         url = get_proxy_run_url(
             service=STREAMS_V1_LOCATION,
             namespace=self.namespace,
@@ -1478,6 +1516,7 @@ class RunClient:
         untar: bool = False,
         overwrite: bool = True,
         show_progress: bool = True,
+        agent: Optional[str] = None,
     ):
         """Uploads a single artifact to the run's artifacts store path.
 
@@ -1488,12 +1527,17 @@ class RunClient:
                  it should be decompressed on the artifacts store.
             overwrite: bool, optional, if the file uploaded should overwrite any previous content.
             show_progress: bool, to show a progress bar.
+            agent: str, optional, uuid reference of an agent to use.
 
         Returns:
             str
         """
         if not self.settings:
             self.refresh_data()
+        if agent:
+            self._reset_agent(agent)
+
+        self._use_agent_host()
 
         params = get_streams_params(connection=self.artifacts_store)
         url = get_proxy_run_url(
@@ -1523,6 +1567,7 @@ class RunClient:
         path: str = "",
         overwrite: bool = True,
         relative_to: Optional[str] = None,
+        agent: Optional[str] = None,
     ):
         """Uploads a full directory to the run's artifacts store path.
 
@@ -1535,7 +1580,7 @@ class RunClient:
             overwrite: bool, optional, if the file uploaded should overwrite any previous content.
             relative_to: str, optional, if the path uploaded is not the current dir,
                  and you want to cancel the relative path.
-
+            agent: str, optional, uuid reference of an agent to use.
         Returns:
             str.
         """
@@ -1555,6 +1600,7 @@ class RunClient:
             path=path or "",
             overwrite=overwrite,
             relative_to=relative_to,
+            agent=agent,
         )
 
     @client_handler(check_no_op=True, check_offline=True)
@@ -1564,6 +1610,7 @@ class RunClient:
         path: str = "",
         overwrite: bool = True,
         relative_to: Optional[str] = None,
+        agent: Optional[str] = None,
     ):
         """Uploads multiple artifacts to the run's artifacts store path.
 
@@ -1573,7 +1620,7 @@ class RunClient:
             overwrite: bool, optional, if the file uploaded should overwrite any previous content.
             relative_to: str, optional, if the path uploaded is not the current dir,
                  and you want to cancel the relative path.
-
+            agent: str, optional, uuid reference of an agent to use.
         Returns:
             str.
         """
@@ -1583,6 +1630,10 @@ class RunClient:
 
         if not self.settings:
             self.refresh_data()
+        if agent:
+            self._reset_agent(agent)
+
+        self._use_agent_host()
 
         params = get_streams_params(connection=self.artifacts_store)
         url = get_proxy_run_url(
@@ -1613,6 +1664,8 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
+
         params = get_streams_params(connection=self.artifacts_store)
         self.client.runs_v1.delete_run_artifact(
             namespace=self.namespace,
@@ -1632,6 +1685,8 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
+
         params = get_streams_params(connection=self.artifacts_store)
         return self.client.runs_v1.delete_run_artifacts(
             namespace=self.namespace,
@@ -1654,6 +1709,8 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
+
         params = get_streams_params(connection=self.artifacts_store)
         return self.client.runs_v1.get_run_artifacts_tree(
             namespace=self.namespace,
@@ -1673,6 +1730,15 @@ class RunClient:
             self.run_uuid,
         )
 
+    @client_handler(check_no_op=True, check_offline=True)
+    def skip(self):
+        """Skip the current run."""
+        self.client.runs_v1.skip_run(
+            self.owner,
+            self.project,
+            self.run_uuid,
+        )
+
     @client_handler(check_no_op=True, check_offline=True)
     def approve(self):
         """Approves the current run if it's pending upload or human approval."""
@@ -2095,9 +2161,9 @@ class RunClient:
         abspath = filepath if is_abs() else os.path.abspath(filepath)
 
         for_patterns = []
-        if getattr(self, "_artifacts_path"):
+        if getattr(self, "_artifacts_path", None):
             for_patterns.append(getattr(self, "_artifacts_path"))
-        if getattr(self, "_store_path"):
+        if getattr(self, "_store_path", None):
             for_patterns.append(getattr(self, "_store_path"))
         context_root = (
             ctx_paths.CONTEXT_OFFLINE_ROOT
@@ -2119,6 +2185,10 @@ class RunClient:
         if not self._has_meta_key("has_events"):
             self.log_meta(has_events=True)
 
+    def _log_has_traces(self):
+        if not self._has_meta_key("has_traces"):
+            self.log_meta(has_traces=True, has_events=True)
+
     def _log_has_metrics(self):
         data = {}
         if not self._has_meta_key("has_metrics"):
@@ -2597,8 +2667,8 @@ class RunClient:
         """
         params = get_query_params(limit=limit, offset=offset, query=query, sort=sort)
         query = params.get("query")
-        query = query + "&" if query else "?"
-        query += "pipeline={}".format(self.run_uuid)
+        query = query + "," if query else ""
+        query += "pipeline:{}".format(self.run_uuid)
         params["query"] = query
 
         return self.client.runs_v1.list_runs(self.owner, self.project, **params)
@@ -2646,6 +2716,21 @@ class RunClient:
             force=force,
         )
 
+    @client_handler(check_no_op=True, check_offline=True)
+    def set_run_edges_lineage(self, run_edges: List[V1RunEdgeLineage]):
+        """
+        Set run edges lineage.
+
+        Args:
+            run_edges: List[V1RunEdgeLineage], list of run edges.
+        """
+        return self.client.runs_v1.set_run_edges_lineage(
+            self.owner,
+            self.project,
+            self.run_uuid,
+            body=V1RunEdgesGraph(edges=run_edges),
+        )
+
     @client_handler(check_no_op=True, check_offline=True)
     def promote_to_artifact_version(
         self,
@@ -2669,7 +2754,7 @@ class RunClient:
             description: str, optional, the version description.
             tags: str or List[str], optional.
             content: str or dict, optional, content/metadata (JSON object) of the version.
-            connection: str, optional, a uuid reference to a connection.
+            connection: str, optional, uuid reference to a connection.
             artifacts: List[str], optional, list of artifacts to highlight(requires passing a run)
             force: bool, optional, to force push, i.e. update if exists.
 
@@ -2812,6 +2897,7 @@ class RunClient:
             config_file.write(
                 orjson_dumps(self.client.sanitize_for_serialization(self.run_data))
             )
+        set_permissions(run_path)
 
         if not self._artifacts_lineage:
             logger.debug("Persist offline run call did not find any lineage data. ")
@@ -2827,6 +2913,7 @@ class RunClient:
                     ]
                 )
             )
+        set_permissions(lineages_path)
 
     @classmethod
     @client_handler(check_no_op=True)
@@ -2835,6 +2922,8 @@ class RunClient:
         path: str,
         run_client: Optional[Union["RunClient", "Run"]] = None,
         reset_project: bool = False,
+        reset_uuid: bool = False,
+        name: Optional[str] = None,
         raise_if_not_found: bool = False,
     ) -> Optional[Union["RunClient", "Run"]]:
         """Loads an offline run from a local path.
@@ -2851,6 +2940,8 @@ class RunClient:
                  from the local run.
             raise_if_not_found: bool, optional, a flag to raise an error if the local path does not
                  contain a persisted run.
+            reset_uuid: bool, optional, a flag to reset the run's uuid.
+            name: str, optional, a name to set for the run.
         """
         run_path = "{}/{}".format(path, ctx_paths.CONTEXT_LOCAL_RUN)
         if not os.path.isfile(run_path):
@@ -2865,6 +2956,10 @@ class RunClient:
             run_config = V1Run(**orjson_loads(config_str))
             owner = run_config.owner
             project = run_config.project
+            if reset_uuid:
+                run_config.uuid = uuid.uuid4().hex
+            if name:
+                run_config.name = name
             if run_client:
                 if reset_project or not owner:
                     owner = run_client.owner
@@ -2923,6 +3018,7 @@ class RunClient:
         path: str,
         upload_artifacts: bool = True,
         clean: bool = False,
+        agent: Optional[str] = None,
     ):
         """Syncs an offline run to Polyaxon's API and artifacts store.
 
@@ -2931,6 +3027,7 @@ class RunClient:
             path: str, root path where the run's metadata & artifacts are stored.
             upload_artifacts: bool, optional, flag to trigger artifacts upload.
             clean: bool, optional, flag to clean local path after pushing the run.
+            agent: str, optional, uuid reference of an agent to use.
         """
         # We ensure that the is_offline is False
         is_offline = self._is_offline
@@ -2963,6 +3060,7 @@ class RunClient:
                 path="/",
                 overwrite=True,
                 relative_to=path,
+                agent=agent,
             )
             logger.info(f"Offline artifacts for run {self.run_data.uuid} uploaded")
 

polyaxon/_compiler/contexts/contexts.py

@@ -55,7 +55,7 @@ def resolve_globals_contexts(
     schedule_at: Optional[datetime] = None,
     started_at: Optional[datetime] = None,
     finished_at: Optional[datetime] = None,
-    duration: Optional[int] = None,
+    duration: Optional[float] = None,
     plugins: Optional[V1Plugins] = None,
     artifacts_store: V1Connection = None,
     cloning_kind: V1CloningKind = None,
@@ -134,7 +134,7 @@ def resolve_contexts(
     schedule_at: Optional[datetime] = None,
     started_at: Optional[datetime] = None,
     finished_at: Optional[datetime] = None,
-    duration: Optional[int] = None,
+    duration: Optional[float] = None,
     cloning_kind: V1CloningKind = None,
     original_uuid: Optional[str] = None,
     is_independent: bool = True,

polyaxon/_compiler/contexts/ray_job.py

@@ -33,7 +33,9 @@ class RayJobContextsManager(BaseContextsManager):
                 connection_by_names=connection_by_names,
             )
 
-        return {
+        data = {
             "head": _get_replica(job.head),
-            "workers": {wn: _get_replica(job.workers[wn]) for wn in job.workers},
         }
+        if job.workers:
+            data["workers"] = {wn: _get_replica(job.workers[wn]) for wn in job.workers}
+        return data

polyaxon/_compiler/resolver/agent.py

@@ -49,8 +49,18 @@ class AgentResolver(BaseSchemaModel):
 
         self.polyaxon_sidecar = agent_config.sidecar or get_default_sidecar_container()
         self.polyaxon_init = agent_config.init or get_default_init_container()
-        self.namespace = agent_config.namespace
-        self.polyaxon_sidecar.monitor_logs = agent_config.is_replica
+        if compiled_operation.namespace:
+            namespaces = agent_config.additional_namespaces or []
+            namespaces.append(agent_config.namespace)
+            if compiled_operation.namespace not in namespaces:
+                raise PolyaxonCompilerError(
+                    "The provided namespace `{}` is not managed by the agent.".format(
+                        compiled_operation.namespace
+                    )
+                )
+            self.namespace = compiled_operation.namespace
+        else:
+            self.namespace = agent_config.namespace
 
     def _resolve_run_connections(
         self, compiled_operation: V1CompiledOperation, agent_config: AgentConfig

polyaxon/_compiler/resolver/runtime.py

@@ -174,7 +174,7 @@ class BaseResolver:
     def resolve_access(self):
         pass
 
-    def resolve_connections(self):
+    def resolve_agent_environment(self):
         agent_env = AgentResolver.construct()
         agent_env.resolve(
             compiled_operation=self.compiled_operation, agent_config=self.agent_config
@@ -281,8 +281,8 @@ class BaseResolver:
         self.apply_operation_contexts()
         self.resolve_io()
         self.resolve_init_refs()
+        self.resolve_agent_environment()
         self.resolve_access()
-        self.resolve_connections()
         self.resolve_artifacts_lineage()
         self.clean_init_refs()
         self.resolve_state()

polyaxon/_contexts/paths.py

@@ -10,7 +10,6 @@ from polyaxon._env_vars.keys import (
     ENV_KEYS_ARTIFACTS_ROOT,
     ENV_KEYS_CONTEXT_ROOT,
     ENV_KEYS_OFFLINE_ROOT,
-    ENV_KEYS_SANDBOX_ROOT,
 )
 
 CONTEXT_RELATED_RUNS = "_related_runs"
@@ -19,7 +18,7 @@ CONTEXT_RELATED_RUNS = "_related_runs"
 CONTEXT_LOCAL_LINEAGES = "lineages.plx.json"
 CONTEXT_LOCAL_CONTENT = "content.plx.json"
 CONTEXT_LOCAL_README = "readme.plx.md"
-CONTEXT_LOCAL_POLYAXONFILE = "polyaxonfile.plx.md"
+CONTEXT_LOCAL_POLYAXONFILE = "polyaxonfile.plx.json"
 CONTEXT_LOCAL_PROJECT = "project.plx.json"
 CONTEXT_LOCAL_RUN = "run.plx.json"
 CONTEXT_LOCAL_VERSION = "version.plx.json"
@@ -53,7 +52,6 @@ CONTEXT_TMP_RUNS_ROOT_FORMAT = os.environ.get(
 )
 CONTEXT_ARCHIVES_ROOT = os.environ.get(ENV_KEYS_ARCHIVES_ROOT, "/tmp/plx/archives")
 CONTEXT_ARTIFACTS_ROOT = os.environ.get(ENV_KEYS_ARTIFACTS_ROOT, "/tmp/plx/artifacts")
-CONTEXT_SANDBOX_ROOT = None
 CONTEXT_OFFLINE_ROOT = os.environ.get(ENV_KEYS_OFFLINE_ROOT, "/tmp/plx/offline")
 CONTEXT_OFFLINE_FORMAT = "{}/{{}}".format(CONTEXT_OFFLINE_ROOT)
 CONTEXT_ARTIFACTS_FORMAT = "{}/{{}}".format(CONTEXT_ARTIFACTS_ROOT)
@@ -84,15 +82,14 @@ def get_offline_path(
 
 
 def mount_sandbox(path: Optional[str] = None):
-    global CONTEXT_SANDBOX_ROOT
     global CONTEXT_OFFLINE_ROOT
     global CONTEXT_ARTIFACTS_ROOT
     global CONTEXT_OFFLINE_FORMAT
     global CONTEXT_ARTIFACTS_FORMAT
 
     path = path or CONTEXT_OFFLINE_ROOT
-    CONTEXT_SANDBOX_ROOT = os.environ.get(ENV_KEYS_SANDBOX_ROOT, path)
-    CONTEXT_OFFLINE_ROOT = CONTEXT_SANDBOX_ROOT
-    CONTEXT_ARTIFACTS_ROOT = CONTEXT_SANDBOX_ROOT
+    CONTEXT_OFFLINE_ROOT = path
+    CONTEXT_ARTIFACTS_ROOT = path
     CONTEXT_OFFLINE_FORMAT = "{}/{{}}".format(CONTEXT_OFFLINE_ROOT)
     CONTEXT_ARTIFACTS_FORMAT = "{}/{{}}".format(CONTEXT_ARTIFACTS_ROOT)
+    return path

polyaxon/_deploy/operators/compose.py

@@ -5,7 +5,7 @@ from polyaxon._env_vars.keys import ENV_KEYS_LOG_LEVEL, ENV_KEYS_SECRET_KEY
 
 
 class ComposeOperator(CmdOperator):
-    CMD = "docker-compose"
+    CMD = "docker compose"
 
     @classmethod
     def params(cls, args):
@@ -112,30 +112,6 @@ class ComposeOperator(CmdOperator):
             )
         return env
 
-    @staticmethod
-    def _generate_intervals(config):
-        template = "{}={}"
-        env = []
-        if config.intervals:
-            env.append(
-                template.format(
-                    "POLYAXON_INTERVALS_RUNS_SCHEDULER", config.intervals.runs_scheduler
-                )
-            )
-            env.append(
-                template.format(
-                    "POLYAXON_INTERVALS_OPERATIONS_DEFAULT_RETRY_DELAY",
-                    config.intervals.operations_default_retry_delay,
-                )
-            )
-            env.append(
-                template.format(
-                    "POLYAXON_INTERVALS_OPERATIONS_MAX_RETRY_DELAY",
-                    config.intervals.operations_max_retry_delay,
-                )
-            )
-        return env
-
     @staticmethod
     def _generate_email(config):
         template = "{}={}"
@@ -195,8 +171,6 @@ class ComposeOperator(CmdOperator):
         if config.allowed_hosts:
             env.append(template.format("POLYAXON_ALLOWED_HOSTS", config.allowed_hosts))
 
-        env += cls._generate_intervals(config)
-
         if config.log_level:
             env.append(template.format(ENV_KEYS_LOG_LEVEL, config.log_level))
         env += cls._generate_postgres(config)