-
Notifications
You must be signed in to change notification settings - Fork 71
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
perf: split metadata/val in objects query #2512
Conversation
Preview this PR with FeatureBee: https://beta.wandb.ai/?betaVersion=0feedb6674a3e2e0cb1531730b534f50f94be4c4 |
@@ -1578,7 +1600,7 @@ def _query_stream( | |||
query: str, | |||
parameters: Dict[str, Any], | |||
column_formats: Optional[Dict[str, Any]] = None, | |||
) -> Iterator[QueryResult]: | |||
) -> Iterator[tuple]: |
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.
This does not return a QueryResult
! its a tuple...
], | ||
row, | ||
list(row) + ["{}"], |
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.
nit: might be nice to comment what this +['{}']
does
adfdb67
to
5661b89
Compare
#2496 but better
Description
This pr:
val_dump
read from the initial select objects queryThis should improve performance on objects with many versions (or lots of data)
version_count
(unused afaict),version_index
which must see all objects.Performance:
Existing queries that crash due to massive memory requirements (
50GB+
) now can be resolved in around 1 second, with a memory requirement less than100MB
total.Testing