Skip to content

Support PipelineTrainer with ServerlessBackend#699

Open
Kovbo wants to merge 3 commits into
mainfrom
serverless-pipeline-trainer-compat
Open

Support PipelineTrainer with ServerlessBackend#699
Kovbo wants to merge 3 commits into
mainfrom
serverless-pipeline-trainer-compat

Conversation

@Kovbo
Copy link
Copy Markdown
Collaborator

@Kovbo Kovbo commented May 28, 2026

Summary

  • extend ServerlessBackend.train to accept the PipelineTrainer train kwargs directly
  • translate supported PipelineTrainer/serverless training fields into ExperimentalTrainingConfig
  • reject unsupported custom loss_fn_config and adam_params explicitly
  • add unit coverage for serverless PipelineTrainer compatibility and payload forwarding

Replaces the relevant serverless PipelineTrainer compatibility work from the stale #623 against current main.

Tests

  • uv run ruff format --check src/art/serverless/backend.py src/art/serverless/client.py tests/unit/test_serverless_pipeline_trainer_compat.py
  • uv run ruff check src/art/serverless/backend.py src/art/serverless/client.py tests/unit/test_serverless_pipeline_trainer_compat.py
  • uv run pytest tests/unit/test_serverless_pipeline_trainer_compat.py -q
  • uv run python -m py_compile src/art/serverless/backend.py src/art/serverless/client.py tests/unit/test_serverless_pipeline_trainer_compat.py

@Kovbo Kovbo force-pushed the serverless-pipeline-trainer-compat branch from 64ddf2c to be350bf Compare May 28, 2026 20:35
@Kovbo Kovbo force-pushed the serverless-pipeline-trainer-compat branch from be350bf to 9366887 Compare May 28, 2026 21:03
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant