nemo-platform-plugin 0.1.0__py3-none-any.whl

nemo_platform/types/jobs/container_spec_param.py

@@ -0,0 +1,37 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Required, TypedDict
+
+from ..._types import SequenceNotStr
+
+__all__ = ["ContainerSpecParam"]
+
+
+class ContainerSpecParam(TypedDict, total=False):
+    """Specification for a container configuration.
+
+    Defines the container image and related configuration for job execution.
+    """
+
+    image: Required[str]
+
+    command: SequenceNotStr[str]
+
+    entrypoint: SequenceNotStr[str]

nemo_platform/types/jobs/cpu_execution_provider.py

@@ -0,0 +1,46 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from typing_extensions import Literal
+
+from ..._models import BaseModel
+from .container_spec import ContainerSpec
+from .compute_resources import ComputeResources
+
+__all__ = ["CPUExecutionProvider"]
+
+
+class CPUExecutionProvider(BaseModel):
+    """CPU-based execution provider.
+
+    Provides configuration for running jobs on CPU resources with
+    resource requests and limits.
+    """
+
+    container: ContainerSpec
+    """Specification for a container configuration.
+
+    Defines the container image and related configuration for job execution.
+    """
+
+    profile: Optional[str] = None
+
+    provider: Optional[Literal["cpu"]] = None
+
+    resources: Optional[ComputeResources] = None
+    """Resource requirements matching k8s ResourceRequirements format."""

nemo_platform/types/jobs/cpu_execution_provider_param.py

@@ -0,0 +1,46 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Literal, Required, TypedDict
+
+from .container_spec_param import ContainerSpecParam
+from .compute_resources_param import ComputeResourcesParam
+
+__all__ = ["CPUExecutionProviderParam"]
+
+
+class CPUExecutionProviderParam(TypedDict, total=False):
+    """CPU-based execution provider.
+
+    Provides configuration for running jobs on CPU resources with
+    resource requests and limits.
+    """
+
+    container: Required[ContainerSpecParam]
+    """Specification for a container configuration.
+
+    Defines the container image and related configuration for job execution.
+    """
+
+    profile: str
+
+    provider: Literal["cpu"]
+
+    resources: ComputeResourcesParam
+    """Resource requirements matching k8s ResourceRequirements format."""

nemo_platform/types/jobs/distributed_gpu_execution_provider.py

@@ -0,0 +1,46 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from typing_extensions import Literal
+
+from ..._models import BaseModel
+from .container_spec import ContainerSpec
+from .compute_resources import ComputeResources
+
+__all__ = ["DistributedGPUExecutionProvider"]
+
+
+class DistributedGPUExecutionProvider(BaseModel):
+    """GPU-based execution provider.
+
+    Provides configuration for running jobs on GPU resources with
+    resource requests and limits.
+    """
+
+    container: ContainerSpec
+    """Specification for a container configuration.
+
+    Defines the container image and related configuration for job execution.
+    """
+
+    profile: Optional[str] = None
+
+    provider: Optional[Literal["gpu_distributed"]] = None
+
+    resources: Optional[ComputeResources] = None
+    """Resource requirements matching k8s ResourceRequirements format."""

nemo_platform/types/jobs/distributed_gpu_execution_provider_param.py

@@ -0,0 +1,46 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Literal, Required, TypedDict
+
+from .container_spec_param import ContainerSpecParam
+from .compute_resources_param import ComputeResourcesParam
+
+__all__ = ["DistributedGPUExecutionProviderParam"]
+
+
+class DistributedGPUExecutionProviderParam(TypedDict, total=False):
+    """GPU-based execution provider.
+
+    Provides configuration for running jobs on GPU resources with
+    resource requests and limits.
+    """
+
+    container: Required[ContainerSpecParam]
+    """Specification for a container configuration.
+
+    Defines the container image and related configuration for job execution.
+    """
+
+    profile: str
+
+    provider: Literal["gpu_distributed"]
+
+    resources: ComputeResourcesParam
+    """Resource requirements matching k8s ResourceRequirements format."""

nemo_platform/types/jobs/docker_job_execution_profile.py

@@ -0,0 +1,43 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from typing_extensions import Literal
+
+from ..._models import BaseModel
+from .docker_job_execution_profile_config import DockerJobExecutionProfileConfig
+
+__all__ = ["DockerJobExecutionProfile"]
+
+
+class DockerJobExecutionProfile(BaseModel):
+    """
+    Execution configuration for a Docker Job.
+    This is used to define the executor type, provider, profile, and any additional configuration
+    required for the executor to run the job on Docker
+    """
+
+    config: DockerJobExecutionProfileConfig
+    """Configuration for Docker Job execution profile."""
+
+    backend: Optional[Literal["docker"]] = None
+
+    profile: Optional[str] = None
+    """The profile name for the executor, e.g., high_priority_a100, low_priority, etc."""
+
+    provider: Optional[str] = None
+    """The compute provider for the executor, e.g., cpu, gpu"""

nemo_platform/types/jobs/docker_job_execution_profile_config.py

@@ -0,0 +1,52 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, Optional
+
+from ..._models import BaseModel
+from .docker_job_network_config import DockerJobNetworkConfig
+from .docker_job_storage_config import DockerJobStorageConfig
+
+__all__ = ["DockerJobExecutionProfileConfig"]
+
+
+class DockerJobExecutionProfileConfig(BaseModel):
+    """Configuration for Docker Job execution profile."""
+
+    cleanup_completed_jobs_immediately: Optional[bool] = None
+
+    env: Optional[Dict[str, str]] = None
+    """Optional env vars applied to all jobs (e.g.
+
+    HOME=/tmp). Keys must not conflict with platform-reserved names. Job steps may
+    override these variables.
+    """
+
+    launcher_tool_path: Optional[str] = None
+    """Path to the jobs launcher tool"""
+
+    networking: Optional[DockerJobNetworkConfig] = None
+    """Docker networking configuration"""
+
+    storage: Optional[DockerJobStorageConfig] = None
+    """Configuration for persistent storage in Docker jobs."""
+
+    ttl_seconds_active: Optional[int] = None
+
+    ttl_seconds_after_finished: Optional[int] = None
+
+    ttl_seconds_before_active: Optional[int] = None

nemo_platform/types/jobs/docker_job_network_config.py

@@ -0,0 +1,27 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+
+from ..._models import BaseModel
+
+__all__ = ["DockerJobNetworkConfig"]
+
+
+class DockerJobNetworkConfig(BaseModel):
+    job_container_network: Optional[str] = None
+    """Docker network for the job container"""

nemo_platform/types/jobs/docker_job_storage_config.py

@@ -0,0 +1,36 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+
+from ..._models import BaseModel
+from .docker_volume_mount import DockerVolumeMount
+
+__all__ = ["DockerJobStorageConfig"]
+
+
+class DockerJobStorageConfig(BaseModel):
+    """Configuration for persistent storage in Docker jobs."""
+
+    additional_volume_mounts: Optional[List[DockerVolumeMount]] = None
+    """List of additional Docker volume mounts for the job"""
+
+    volume_name: Optional[str] = None
+    """Name of the Docker volume for persistent storage"""
+
+    volume_permissions_image: Optional[str] = None
+    """Docker image used to set permissions on the volume"""

nemo_platform/types/jobs/docker_volume_mount.py

@@ -0,0 +1,47 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, Optional
+from typing_extensions import Literal
+
+from ..._models import BaseModel
+
+__all__ = ["DockerVolumeMount"]
+
+
+class DockerVolumeMount(BaseModel):
+    mount_path: str
+    """Path inside the container where the volume will be mounted"""
+
+    volume_name: str
+    """Name of the Docker volume to mount"""
+
+    allow_create_volume: Optional[bool] = None
+    """
+    Whether to allow the creation of the volume if it does not exist (default:
+    false).
+    """
+
+    kind: Optional[Literal["volume", "tmpfs"]] = None
+    """Type of the Docker volume to mount.
+
+    Options are 'volume' or 'tmpfs' (default: 'volume'). tmpfs volumes are only
+    supported on Linux hosts.
+    """
+
+    options: Optional[Dict[str, object]] = None
+    """Additional options for the volume"""

nemo_platform/types/jobs/e2e_job_execution_profile.py

@@ -0,0 +1,43 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from typing_extensions import Literal
+
+from ..._models import BaseModel
+from .job_execution_profile_config import JobExecutionProfileConfig
+
+__all__ = ["E2EJobExecutionProfile"]
+
+
+class E2EJobExecutionProfile(BaseModel):
+    """
+    Execution configuration for E2E testing.
+    This backend auto-completes jobs without actually running containers,
+    making tests fast and deterministic.
+    """
+
+    backend: Optional[Literal["e2e"]] = None
+
+    config: Optional[JobExecutionProfileConfig] = None
+    """Configuration for the e2e test executor"""
+
+    profile: Optional[str] = None
+    """The profile name for the executor, e.g., high_priority_a100, low_priority, etc."""
+
+    provider: Optional[str] = None
+    """The compute provider for the executor, e.g., cpu, gpu"""

nemo_platform/types/jobs/gpu_execution_provider.py

@@ -0,0 +1,46 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from typing_extensions import Literal
+
+from ..._models import BaseModel
+from .container_spec import ContainerSpec
+from .compute_resources import ComputeResources
+
+__all__ = ["GPUExecutionProvider"]
+
+
+class GPUExecutionProvider(BaseModel):
+    """GPU-based execution provider.
+
+    Provides configuration for running jobs on GPU resources with
+    resource requests and limits.
+    """
+
+    container: ContainerSpec
+    """Specification for a container configuration.
+
+    Defines the container image and related configuration for job execution.
+    """
+
+    profile: Optional[str] = None
+
+    provider: Optional[Literal["gpu"]] = None
+
+    resources: Optional[ComputeResources] = None
+    """Resource requirements matching k8s ResourceRequirements format."""

nemo_platform/types/jobs/gpu_execution_provider_param.py

@@ -0,0 +1,46 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Literal, Required, TypedDict
+
+from .container_spec_param import ContainerSpecParam
+from .compute_resources_param import ComputeResourcesParam
+
+__all__ = ["GPUExecutionProviderParam"]
+
+
+class GPUExecutionProviderParam(TypedDict, total=False):
+    """GPU-based execution provider.
+
+    Provides configuration for running jobs on GPU resources with
+    resource requests and limits.
+    """
+
+    container: Required[ContainerSpecParam]
+    """Specification for a container configuration.
+
+    Defines the container image and related configuration for job execution.
+    """
+
+    profile: str
+
+    provider: Literal["gpu"]
+
+    resources: ComputeResourcesParam
+    """Resource requirements matching k8s ResourceRequirements format."""

nemo_platform/types/jobs/image_pull_secret.py

@@ -0,0 +1,27 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from ..._models import BaseModel
+
+__all__ = ["ImagePullSecret"]
+
+
+class ImagePullSecret(BaseModel):
+    """Kubernetes image pull secret reference."""
+
+    name: str
+    """Kubernetes Secret name for pulling images"""

nemo_platform/types/jobs/job_create_params.py

@@ -0,0 +1,46 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict
+from typing_extensions import Required, TypedDict
+
+from .platform_job_spec_param import PlatformJobSpecParam
+
+__all__ = ["JobCreateParams"]
+
+
+class JobCreateParams(TypedDict, total=False):
+    workspace: str
+
+    platform_spec: Required[PlatformJobSpecParam]
+    """Specification for a platform job, containing steps and secrets."""
+
+    source: Required[str]
+
+    spec: Required[Dict[str, object]]
+
+    custom_fields: Dict[str, object]
+
+    description: str
+
+    name: str
+
+    ownership: Dict[str, object]
+
+    project: str