Skip to content

Small modelbench fixes#978

Merged
wpietri merged 1 commit into
mainfrom
small_running_fixes
Apr 14, 2025
Merged

Small modelbench fixes#978
wpietri merged 1 commit into
mainfrom
small_running_fixes

Conversation

@wpietri
Copy link
Copy Markdown
Contributor

@wpietri wpietri commented Apr 14, 2025

Make together retry more. Allow changing number of threads when running benchmarks. Speed up retry tests and other sleeping tests.

Allow changing number of threads when running benchmarks.
Speed up `retry` tests and other sleeping tests.
@wpietri wpietri requested a review from a team as a code owner April 14, 2025 20:29
@wpietri wpietri requested a review from bkorycki April 14, 2025 20:30
@wpietri wpietri temporarily deployed to Scheduled Testing April 14, 2025 20:30 — with GitHub Actions Inactive
@wpietri wpietri temporarily deployed to Scheduled Testing April 14, 2025 20:30 — with GitHub Actions Inactive
@wpietri wpietri temporarily deployed to Scheduled Testing April 14, 2025 20:30 — with GitHub Actions Inactive
@github-actions
Copy link
Copy Markdown

MLCommons CLA bot All contributors have signed the MLCommons CLA ✍️ ✅

Copy link
Copy Markdown
Contributor

@rogthefrog rogthefrog left a comment

Choose a reason for hiding this comment

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

👍🏻


class AnnotatorWorkers(CachingPipe):
def __init__(self, annotators: dict[str, Annotator], workers=None, cache_path=None):
self.sleep_time = 10
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Should it be a class-level setting?

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

Probably not, as the use for it is to change one instance for a test.


class PromptSutWorkers(CachingPipe):
def __init__(self, suts: dict[str, SUT], sut_options: Optional[SUTOptions] = None, workers=None, cache_path=None):
self.sleep_time = 10
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Same question as above.

Copy link
Copy Markdown
Contributor

@bkorycki bkorycki left a comment

Choose a reason for hiding this comment

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

looks good!

Copy link
Copy Markdown
Collaborator

@dhosterman dhosterman left a comment

Choose a reason for hiding this comment

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

👍

@wpietri wpietri merged commit 40bbaba into main Apr 14, 2025
4 checks passed
@github-actions github-actions Bot locked and limited conversation to collaborators Apr 14, 2025
@wpietri wpietri deleted the small_running_fixes branch January 16, 2026 14:08
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants