-
Notifications
You must be signed in to change notification settings - Fork 151
Add support for Experiment tracking in Model Registry, fixes #1224 #1318
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
Merged
google-oss-prow
merged 54 commits into
kubeflow:main
from
dhirajsb:feat/experiment-tracking-api
Aug 12, 2025
Merged
Changes from 38 commits
Commits
Show all changes
54 commits
Select commit
Hold shift + click to select a range
334cdd4
feat: initial version of experiments and runs API
dhirajsb c0ad3cd
feat: experiments and runs initial implementation (wip)
dhirajsb e908bfb
fix: fixed failing unit tests for experiments and runs
dhirajsb 3111574
fix: added experiment and experimentrun tests
dhirajsb 4324ba7
feat: added DataSet, Metric, and Parameter types
dhirajsb dc1c492
feat: added implementatio of DataSet, Metric, and Param, including se…
dhirajsb c7d079c
fix: replace int properties for timestamps with string because mlmd t…
dhirajsb e1bd49d
feat: add support for artifactType query param to filter artifact typ…
dhirajsb 60c762f
fix: add metrics history endpoint and metric history storage for expe…
dhirajsb 0f311d0
fix: fix artifactType query param type in generated service
dhirajsb 122a529
fix: fix go lint error in unit test
dhirajsb 17d3f77
fix: filter out metric history from artifacts endpoints
dhirajsb 18b8ec1
fix: fix metric history name to use last update time to avoid name co…
dhirajsb f5891ed
feat: add filterQuery param on all context types to search by propert…
dhirajsb 745db32
feat: initial version of experiment tracking implemented on embedmd, …
dhirajsb c9df459
feat: add support for filterQuery parameter for all ListResponse endp…
dhirajsb 2270de9
fix: add support for stepIds query parameter in embedmd datastore
dhirajsb 55b5b50
feat: refactor embedmd db service to use generic repository implement…
dhirajsb c6a3d25
fix: add support for artifactType query parameter for embedmd datastore
dhirajsb aceb613
fix: use mysql 8.3 in unit tests
dhirajsb 72e6c48
fix: refactor name mapping and default name handling in embedmd datas…
dhirajsb fe74251
feat: support updating metrics and parameters by name, fix ignoring m…
dhirajsb 7a95c25
fix: add missing generated openapi python client files for PR github …
dhirajsb 317d159
fix: fix failing shared db tests
dhirajsb 4eea254
fix: add support for metric and parameter description, add missing ty…
dhirajsb d6a8582
chore: update files from main
Al-Pragliola 79b1316
Merge branch 'main' into feat/experiment-tracking-api
Al-Pragliola e3d7853
fix: added missing godoc comments in pkg/api/api.go
dhirajsb cd8cbba
fix: replace ambiguous ArtifactListReponse return type from GetExperi…
dhirajsb d4bb6e8
fix: fixed incorrect artifactType in dataset response, added tests to…
dhirajsb 0d972cf
feat: add validation for endTimeSinceEpoch property on experiment run…
dhirajsb 04c0d66
Replace value type validation map with a switch in query_translator.go
dhirajsb 25fb7b5
fix: add service e2e tests for filterQuery, fix name query param hand…
dhirajsb eddc5e0
chore: code cleanup, replace interface{} with any, added vetting for …
dhirajsb 34c3ccc
chore: added flag vF for fixed string grep exclude
dhirajsb 1af2a6e
fix: copied orderby and parameters back to registry and catalog to ha…
dhirajsb 41d24c1
fix: fixed mlmd query translator handling of escaped backslashes
dhirajsb 2c67e6f
chore: add test to verify parseCustomPropertyField won't panic with a…
dhirajsb 57e02fa
fix: sync generated python client code
dhirajsb a7ffa31
Merge branch 'main' into feat/experiment-tracking-api
Al-Pragliola fc72ee0
fix: readiness probe tests and new types
Al-Pragliola 9e2ee3c
chore: refactor readiness_test
Al-Pragliola 5a54028
fix: ensure parentResourceId is used to filter resource lookup by par…
dhirajsb 76d7bb1
fix: throw an error if a metric value is missing, add test to validate
dhirajsb fe06cc2
fix: fix http status error code for invalid ids
dhirajsb dde506f
fix: more id validation, fixed filterQuery passing to DB layer
dhirajsb 540eb49
fix: fix failing unit test
dhirajsb 48d0f97
fix: validate experiment id when listing runs
dhirajsb e605f59
fix: fix failing validation test after fixing http status codes
dhirajsb 1d49337
fix: avoid duplicate key errors if externalid is set in metric when c…
dhirajsb b98473b
fix: add fuzzer tests for experiment runs and new artifact types
dhirajsb ce403b2
chore: code cleanup and format fuzzer tests
dhirajsb f99d1a5
fix: log error in fuzzer test
dhirajsb 05ea1ce
fix: handle null artifact names correctly on create
dhirajsb File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Uh oh!
There was an error while loading. Please reload this page.