Skip to content

Conversation

@dorotat-nv
Copy link
Collaborator

@dorotat-nv dorotat-nv commented Sep 5, 2025

Description

Fixed installation of the subquadratic package in the command for Evo2 pretrain partial-convergence benchmarks

Type of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Refactor
  • Documentation update
  • Other (please describe):

CI Pipeline Configuration

Configure CI behavior by applying the relevant labels:

Note

By default, the notebooks validation tests are skipped unless explicitly enabled.

Authorizing CI Runs

We use copy-pr-bot to manage authorization of CI
runs on NVIDIA's compute resources.

  • If a pull request is opened by a trusted user and contains only trusted changes, the pull request's code will
    automatically be copied to a pull-request/ prefixed branch in the source repository (e.g. pull-request/123)
  • If a pull request is opened by an untrusted user or contains untrusted changes, an NVIDIA org member must leave an
    /ok to test comment on the pull request to trigger CI. This will need to be done for each new commit.

Usage

# TODO: Add code snippet

Pre-submit Checklist

  • I have tested these changes locally
  • I have updated the documentation accordingly
  • I have added/updated tests as needed
  • All existing tests pass successfully

Summary by CodeRabbit

  • Refactor
    • Renamed public configuration key from artefacts_url to pckg_url. Update any custom configs or scripts accordingly.
  • Chores
    • Updated package installation to use an authenticated index URL with a fallback to the public index for improved reliability.
    • Aligned default URL format to match the new pckg_url key.

@dorotat-nv dorotat-nv requested a review from pstjohn as a code owner September 5, 2025 09:44
@coderabbitai
Copy link
Contributor

coderabbitai bot commented Sep 5, 2025

Walkthrough

Renamed the config key artefacts_url to pckg_url in evo2_pretrain.yaml, updated its default value, and modified the pip install command to construct an authenticated index URL using pckg_url with an extra index pointing to PyPI.

Changes

Cohort / File(s) Summary
CI benchmark config update
ci/benchmarks/partial-conv/evo2_pretrain.yaml
Renamed key_segments.script_args keys artefacts_url -> pckg_url; updated default URL value; changed pip install to use --index-url https://oauth2:$JET_GITLAB_TOKEN@${pckg_url} and add --extra-index-url https://pypi.org/simple/.

Sequence Diagram(s)

sequenceDiagram
  autonumber
  participant CI as CI Runner
  participant SH as Shell Script
  participant Pip as pip
  participant GL as GitLab Package Registry
  participant PYPI as PyPI

  CI->>SH: Run evo2_pretrain job
  SH->>Pip: pip install ${file_name_wheel} --index-url https://oauth2:$JET_GITLAB_TOKEN@${pckg_url} --extra-index-url https://pypi.org/simple/
  Pip->>GL: Authenticate with oauth2 token and fetch package
  alt Package not found in GitLab
    Pip->>PYPI: Fallback to PyPI simple index
  end
  Pip-->>SH: Installation result
  SH-->>CI: Job continues
Loading

Estimated code review effort

🎯 2 (Simple) | ⏱️ ~10 minutes

Poem

I hop through YAML fields at night,
Renamed a key—now tidy, right?
A token taps the registry door,
Pip finds the treats it’s searching for.
If not, to PyPI I bound with glee—
Carrots cached, dependencies free! 🥕

✨ Finishing Touches
🧪 Generate unit tests
  • Create PR with unit tests
  • Post copyable unit tests in a comment
  • Commit unit tests in branch dorotat/fix-jet-evo2-pretrain

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share
🪧 Tips

Chat

There are 3 ways to chat with CodeRabbit:

  • Review comments: Directly reply to a review comment made by CodeRabbit. Example:
    • I pushed a fix in commit <commit_id>, please review it.
    • Open a follow-up GitHub issue for this discussion.
  • Files and specific lines of code (under the "Files changed" tab): Tag @coderabbitai in a new review comment at the desired location with your query.
  • PR comments: Tag @coderabbitai in a new PR comment to ask questions about the PR branch. For the best results, please provide a very specific query, as very limited context is provided in this mode. Examples:
    • @coderabbitai gather interesting stats about this repository and render them as a table. Additionally, render a pie chart showing the language distribution in the codebase.
    • @coderabbitai read the files in the src/scheduler package and generate a class diagram using mermaid and a README in the markdown format.

Support

Need help? Create a ticket on our support page for assistance with any issues or questions.

CodeRabbit Commands (Invoked using PR/Issue comments)

Type @coderabbitai help to get the list of available commands.

Other keywords and placeholders

  • Add @coderabbitai ignore or @coderabbit ignore anywhere in the PR description to prevent this PR from being reviewed.
  • Add @coderabbitai summary to generate the high-level summary at a specific location in the PR description.
  • Add @coderabbitai anywhere in the PR title to generate the title automatically.

CodeRabbit Configuration File (.coderabbit.yaml)

  • You can programmatically configure CodeRabbit by adding a .coderabbit.yaml file to the root of your repository.
  • Please see the configuration documentation for more information.
  • If your editor has YAML language server enabled, you can add the path at the top of this file to enable auto-completion and validation: # yaml-language-server: $schema=https://coderabbit.ai/integrations/schema.v2.json

Status, Documentation and Community

  • Visit our Status Page to check the current availability of CodeRabbit.
  • Visit our Documentation for detailed information on how to use CodeRabbit.
  • Join our Discord Community to get help, request features, and share feedback.
  • Follow us on X/Twitter for updates and announcements.

@dorotat-nv
Copy link
Collaborator Author

/ok to test 5ff044a

@dorotat-nv dorotat-nv enabled auto-merge September 5, 2025 09:45
Copy link
Contributor

@coderabbitai coderabbitai bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 1

🧹 Nitpick comments (2)
ci/benchmarks/partial-conv/evo2_pretrain.yaml (2)

18-18: Clarify the expected format of pckg_url (host/path vs full URL).

Right now the script prepends https://oauth2:$JET_GITLAB_TOKEN@ to ${pckg_url}. If someone overrides pckg_url with a value that already includes https://, you’ll end up with an invalid URL. Either:

  • Document that pckg_url must be “host/path only” (no scheme), or
  • Accept both by sanitizing the scheme in the script (see next comment’s diff).

Would you like me to update this with an inline comment next to pckg_url stating “no scheme, trailing slash required”?


10-10: Key rename correct; no dangling artefacts_url references. Grep confirms only pckg_url is used in ci/benchmarks/partial-conv/evo2_pretrain.yaml (no artefacts_url remains). Optional: consider renaming to pkg_url or package_index_url for clarity.

📜 Review details

Configuration used: CodeRabbit UI

Review profile: CHILL

Plan: Pro

💡 Knowledge Base configuration:

  • MCP integration is disabled by default for public repositories
  • Jira integration is disabled by default for public repositories
  • Linear integration is disabled by default for public repositories

You can enable these sources in your CodeRabbit configuration.

📥 Commits

Reviewing files that changed from the base of the PR and between d91788a and 5ff044a.

📒 Files selected for processing (1)
  • ci/benchmarks/partial-conv/evo2_pretrain.yaml (2 hunks)
⏰ Context from checks skipped due to timeout of 90000ms. You can increase the timeout in your CodeRabbit configuration to a maximum of 15 minutes (900000ms). (1)
  • GitHub Check: Analyze (rust)

@codecov-commenter
Copy link

codecov-commenter commented Sep 5, 2025

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 80.62%. Comparing base (d91788a) to head (5ff044a).
⚠️ Report is 4 commits behind head on main.

Additional details and impacted files
@@           Coverage Diff           @@
##             main    #1115   +/-   ##
=======================================
  Coverage   80.62%   80.62%           
=======================================
  Files         157      157           
  Lines       11079    11079           
=======================================
  Hits         8932     8932           
  Misses       2147     2147           

@dorotat-nv
Copy link
Collaborator Author

/ok to test f7c8f58

@dorotat-nv dorotat-nv added this pull request to the merge queue Sep 8, 2025
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Sep 8, 2025
@dorotat-nv dorotat-nv added this pull request to the merge queue Sep 8, 2025
Merged via the queue into main with commit a1e2340 Sep 8, 2025
19 checks passed
@dorotat-nv dorotat-nv deleted the dorotat/fix-jet-evo2-pretrain branch September 8, 2025 16:04
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.

5 participants