Skip to content

Commit f45661e

Browse files
committed
Rename parameters to match Management API
1 parent d6887dc commit f45661e

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

singlestoredb/management/export.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def __repr__(self) -> str:
2828
return str(self)
2929

3030
@abc.abstractmethod
31-
def to_storage_location(self) -> Dict[str, Any]:
31+
def to_storage_info(self) -> Dict[str, Any]:
3232
raise NotImplementedError
3333

3434
@classmethod
@@ -63,7 +63,7 @@ def __init__(self, region: str, storage_base_url: str):
6363
self.storage_base_url = storage_base_url
6464
self._manager: Optional[WorkspaceManager] = None
6565

66-
def to_storage_location(self) -> Dict[str, Any]:
66+
def to_storage_info(self) -> Dict[str, Any]:
6767
return dict(
6868
storageBaseURL=self.storage_base_url,
6969
storageRegion=self.region,
@@ -167,7 +167,7 @@ def from_config_and_creds(
167167
def to_catalog_info(self) -> Dict[str, Any]:
168168
"""Return a catalog info dictionary."""
169169
return dict(
170-
catalogSource=self.catalog_type,
170+
catalogType=self.catalog_type,
171171
tableFormat=self.table_format,
172172
glueRegion=self.region,
173173
glueCatalogID=self.catalog_id,
@@ -262,7 +262,7 @@ def start(self, tags: Optional[List[str]] = None) -> 'ExportStatus':
262262
json=dict(
263263
databaseName=self.database,
264264
tableName=self.table,
265-
storageLocation=self.storage_link.to_storage_location(),
265+
storageInfo=self.storage_link.to_storage_info(),
266266
catalogInfo=self.catalog.to_catalog_info(),
267267
),
268268
)

0 commit comments

Comments
 (0)