File tree Expand file tree Collapse file tree 2 files changed +9
-3
lines changed
Expand file tree Collapse file tree 2 files changed +9
-3
lines changed Original file line number Diff line number Diff line change @@ -753,14 +753,17 @@ def export_timeseries(
753753
754754 # User data access
755755
756- def get_user_data_credentials (self ) -> UserDataCredentials :
756+ def get_user_data_credentials (
757+ self , organization_id : UUID , workspace_id : UUID
758+ ) -> UserDataCredentials :
757759 """
758760 Retrieves user data credentials for secondary storage access.
759761
760762 Returns:
761763 The user data credentials object.
762764 """
763- response = self ._request ("GET" , "/user-data/credentials" )
765+ params = {"org_id" : str (organization_id ), "workspace_id" : str (workspace_id )}
766+ response = self ._request ("GET" , "/user-data/credentials" , params = params )
764767 return UserDataCredentials (** response .json ())
765768
766769 # Container registry access
Original file line number Diff line number Diff line change @@ -687,8 +687,11 @@ def initialize(self) -> None:
687687 if self ._api is not None :
688688 api = self ._api
689689 self ._credential_manager = CredentialManager (
690- credential_fetcher = lambda : api .get_user_data_credentials ()
690+ credential_fetcher = lambda : api .get_user_data_credentials (
691+ self ._organization .id , self ._workspace .id
692+ )
691693 )
694+
692695 self ._credential_manager .initialize ()
693696
694697 self ._fs = self ._credential_manager .get_filesystem ()
You can’t perform that action at this time.
0 commit comments