skill refinement

Signed-off-by: Sam Oluwalana <[email protected]>

Author: soluwalana

docs/set-up/config-reference.md

@@ -409,6 +409,8 @@ jobs:
   reconcile_interval_seconds: 2
   # Interval in seconds for the job scheduler to run | default: 5
   schedule_interval_seconds: 5
+  # Register the subprocess/default execution profile. When unset, defaults to true for docker/none runtimes and false for kubernetes.
+  enable_subprocess_executor:
 ```
 
 ### `models`

packages/nemo_platform_plugin/src/nemo_platform_plugin/discovery.py

@@ -52,8 +52,8 @@
 if TYPE_CHECKING:
     from nemo_platform_plugin.cli import NemoCLI
     from nemo_platform_plugin.controller import NemoController
-    from nemo_platform_plugin.function import NemoFunction
     from nemo_platform_plugin.customization_contributor import CustomizationContributor
+    from nemo_platform_plugin.function import NemoFunction
     from nemo_platform_plugin.inference_middleware import NemoInferenceMiddleware
     from nemo_platform_plugin.job import NemoJob
     from nemo_platform_plugin.seed import NemoSeedJob
@@ -501,7 +501,6 @@ def discover_customization_contributors() -> dict[str, CustomizationContributor]
     instance. Broken contributors are skipped with a warning (same fault isolation as
     :func:`discover`).
     """
-    from nemo_platform_plugin.customization_contributor import CustomizationContributor
 
     result: dict[str, CustomizationContributor] = {}
 

packages/nemo_platform_plugin/tests/test_authz.py

@@ -4,7 +4,6 @@
 from __future__ import annotations
 
 import pytest
-
 from nemo_platform_plugin.authz import AuthzContribution, authz_for_workspace_job_collection
 from nemo_platform_plugin.authz_discovery import _collect_from_plugin_surface, discover_authz_contributions
 from nemo_platform_plugin.service import NemoService

plugins/nemo-automodel/src/nemo_automodel_plugin/config.py

@@ -5,8 +5,6 @@
 
 from __future__ import annotations
 
-import os
-
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
 

plugins/nemo-automodel/src/nemo_automodel_plugin/contributor.py

@@ -56,14 +56,15 @@ async def healthz() -> dict[str, str]:
         ]
 
     def get_cli(self) -> typer.Typer:
-        from nemo_automodel_plugin.cli.inputs import apply_automodel_job_cli_overrides
         from nemo_platform_plugin.commands import (
             _add_explain_command,
             _add_run_command,
             _add_submit_command,
         )
         from nemo_platform_plugin.scheduler import NemoJobScheduler
 
+        from nemo_automodel_plugin.cli.inputs import apply_automodel_job_cli_overrides
+
         app = typer.Typer(
             name=self.name,
             help="Automodel training jobs (SFT, distillation).",

plugins/nemo-automodel/src/nemo_automodel_plugin/sdk/http_utils.py

@@ -8,9 +8,10 @@
 from typing import Any
 from urllib.parse import quote, urljoin
 
-from nemo_automodel_plugin.schema import AutomodelJobInput
 from nemo_platform import AsyncNeMoPlatform, NeMoPlatform
 
+from nemo_automodel_plugin.schema import AutomodelJobInput
+
 PlatformClient = NeMoPlatform | AsyncNeMoPlatform
 
 _API_PREFIX = "/apis/customization"

plugins/nemo-automodel/src/nemo_automodel_plugin/sdk/job_resources.py

@@ -8,11 +8,11 @@
 from typing import Any
 from urllib.parse import quote
 
-from nemo_automodel_plugin.sdk import http_utils
-from nemo_platform import AsyncNeMoPlatform, NeMoPlatform
 from nemo_platform_plugin.jobs.schemas import PlatformJobStatusResponse
 from pydantic import BaseModel
 
+from nemo_automodel_plugin.sdk import http_utils
+
 
 class AutomodelJobRecord(BaseModel):
     """Minimal job record returned by the customization Automodel jobs API."""

plugins/nemo-automodel/src/nemo_automodel_plugin/sdk/resources.py

@@ -7,14 +7,15 @@
 
 from typing import Any
 
+from nemo_platform import AsyncNeMoPlatform, NeMoPlatform
+
 from nemo_automodel_plugin.schema import AutomodelJobInput
 from nemo_automodel_plugin.sdk import http_utils
 from nemo_automodel_plugin.sdk.job_resources import (
     AsyncAutomodelJobResource,
     AutomodelJobRecord,
     AutomodelJobResource,
 )
-from nemo_platform import AsyncNeMoPlatform, NeMoPlatform
 
 
 class AutomodelJobsResource:

plugins/nemo-automodel/src/nemo_automodel_plugin/transform.py

@@ -8,14 +8,14 @@
 import uuid
 from typing import TYPE_CHECKING
 
+from nmp.automodel.platform_client import check_dataset_access, fetch_model_entity
+from nmp.common.entities.utils import parse_entity_ref
+
 from nemo_automodel_plugin.schema import (
     AutomodelJobInput,
     AutomodelJobOutput,
-    OutputRequest,
     OutputResponse,
 )
-from nmp.automodel.platform_client import check_dataset_access, fetch_model_entity
-from nmp.common.entities.utils import parse_entity_ref
 
 if TYPE_CHECKING:
     from nemo_platform import AsyncNeMoPlatform

plugins/nemo-automodel/tests/test_cli.py

@@ -122,5 +122,3 @@ def test_cli_expose_input_and_output_schemas() -> None:
     assert "input_spec_schema" in payload
     assert "spec_schema" in payload
     assert "/automodel/jobs" in payload["endpoint"]
-
-