Skip to content

Commit 80c07a1

Browse files
committed
Remove Jobs Stainless SDK dependency
Signed-off-by: Ryan S <267728323+ironcommit@users.noreply.github.com>
1 parent d5bee2b commit 80c07a1

37 files changed

Lines changed: 990 additions & 435 deletions

File tree

e2e/test_evaluator_plugin.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -830,9 +830,10 @@ def test_gym_agent_evaluate_job_completes(
830830

831831
# Download the results archive to a tmp path.
832832
archive_path = tmp_path / "agent-eval-results.tar.gz"
833-
evaluator_sdk.jobs.results.download(
834-
"agent-eval-results", job=job_name, workspace=evaluator_workspace
835-
).write_to_file(archive_path)
833+
archive = client_from_platform(evaluator_sdk, JobsClient).download_job_result(
834+
name="agent-eval-results", job=job_name, workspace=evaluator_workspace
835+
)
836+
archive_path.write_bytes(archive.read())
836837

837838
# Extract the trials.jsonl file from the archive.
838839
extract_dir = tmp_path / "agent-eval-results"

packages/nemo_platform_ext/src/nemo_platform_ext/cli/commands/api/jobs/__init__.py

Lines changed: 48 additions & 25 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

packages/nemo_platform_ext/src/nemo_platform_ext/cli/commands/api/jobs/results.py

Lines changed: 18 additions & 6 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

packages/nemo_platform_ext/src/nemo_platform_ext/cli/commands/api/jobs/steps.py

Lines changed: 29 additions & 9 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)