Skip to content

feat(feature_store): adds athena workgroup to feature store DatasetBuilder - #6261

Open
johnrc90 wants to merge 2 commits into
aws:masterfrom
johnrc90:feature-add-workgroup-to-dataset-builder
Open

johnrc90 wants to merge 2 commits into
aws:masterfrom
johnrc90:feature-add-workgroup-to-dataset-builder

Conversation

@johnrc90

@johnrc90 johnrc90 commented Sep 11, 2026

Copy link
Copy Markdown

*Issue #, if available: #3176 (comment)

Description of changes:

There are scenarios where the user might want to select a specific Athena Workgroup (organization requirements, Sagemaker Unified Studio) to use the feature store helper DatasetBuilder.

The feature_store.feature_utils.start_query_execution already had the workgroup parameter, but the other feature_store functions that called it never used it.

This PR adds the workgroup parameter to DatasetBuilder.create and the following functions that would use it (DatasetBuilder._run_query -> feature_utils.run_athena_query -> feature_utils.start_query_execution)

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@johnrc90 johnrc90 changed the title feature: adds athena workgroup to feature store DatasetBuilder feat(feature_store): adds athena workgroup to feature store DatasetBuilder Sep 21, 2026

This branch is waiting to be deployed

1 waiting deployment
manual-approval 8513cf56 Waiting Sep 14, 2026 by johnrc90 via wait-for-approval #340
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant