-
Notifications
You must be signed in to change notification settings - Fork 356
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
Make sure pruning does prune #1014
Conversation
@@ -6,7 +6,7 @@ | |||
import pickle |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
changes to this files are unrelated, just a convention update since I stumbled on it in the process
@@ -269,8 +275,9 @@ def __call__(self, archive: Archive[MultiValue]) -> Archive[MultiValue]: | |||
new_archive.bytesdict = { | |||
b: v | |||
for b, v in archive.bytesdict.items() | |||
if any(v.get_estimation(n) <= quantiles[n] for n in names) | |||
} | |||
if any(v.get_estimation(n) < quantiles[n] for n in names) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
the strict comparison is the big change that make it work.
@@ -13,4 +13,4 @@ | |||
__all__ = ["optimizers", "families", "callbacks", "p", "typing"] | |||
|
|||
|
|||
__version__ = "0.4.2.post5" | |||
__version__ = "0.4.2.post6" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
new version because this is very impactful
Types of changes
Motivation and Context / Related issue
This may solve #1013
The issue came from repeated values which may not be pruned, making the pruning mechanism be called way too often
How Has This Been Tested (if it applies)
Added a test to make sure pruning was not called too often, in the constant return case (worst case scenario)
Checklist