Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adding in CPU time to wherever time is defined as variable "cpu_time" #1173

Merged
merged 5 commits into from
Nov 29, 2024

Conversation

thijssnelleman
Copy link

Changed the TrialValue to hold CPU time and have SMBO use this value to calculate its budgets

  • Same behaviour as the original 'time' variable, except in its calculation tries to estimate CPU time with process_time instead.
  • Added in to TrialValue object, RunHistory and SMBO so it can be recorded and correctly calculate the remaining CPU budget

Pytest are passing (on my machine)

Copy link
Collaborator

@benjamc benjamc left a comment

Choose a reason for hiding this comment

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

is there any way to test the new cpu return? if yes, please add a little test

otherwise, codewise the PR LGTM, it only needs a bit more documentation.
You can basically copy/paste the description for cpu_time everywhere.

CHANGELOG.md Outdated Show resolved Hide resolved
smac/main/smbo.py Outdated Show resolved Hide resolved
smac/runhistory/dataclasses.py Show resolved Hide resolved
smac/runhistory/runhistory.py Show resolved Hide resolved
smac/runhistory/runhistory.py Show resolved Hide resolved
smac/runner/target_function_runner.py Show resolved Hide resolved
smac/runner/abstract_runner.py Outdated Show resolved Hide resolved
smac/runner/target_function_script_runner.py Show resolved Hide resolved
@benjamc benjamc self-assigned this Nov 27, 2024
@benjamc benjamc added this to the v2.3 milestone Nov 27, 2024
@thijssnelleman
Copy link
Author

Added in CPU time tests in target function runner tests

@benjamc benjamc merged commit 000dd2d into automl:development Nov 29, 2024
11 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
Status: Done
Development

Successfully merging this pull request may close these issues.

2 participants