model files are not staged on dremio/athena but executed directly

This commit is contained in:
djudjuu
2025-12-11 17:30:38 +01:00
parent 18d46a9db3
commit 83ee00ddb3
2 changed files with 4 additions and 4 deletions

View File

@@ -131,10 +131,10 @@ class athena(Destination[AthenaClientConfiguration, "AthenaClient"]):
caps = DestinationCapabilitiesContext()
# athena only supports loading from staged files on s3 for now
caps.preferred_loader_file_format = None
caps.supported_loader_file_formats = []
caps.supported_loader_file_formats = ["model"]
caps.supported_table_formats = ["iceberg", "hive"]
caps.preferred_staging_file_format = "parquet"
caps.supported_staging_file_formats = ["parquet", "model"]
caps.supported_staging_file_formats = ["parquet"]
caps.type_mapper = AthenaTypeMapper
# athena is storing all identifiers in lower case and is case insensitive

View File

@@ -87,9 +87,9 @@ class dremio(Destination[DremioClientConfiguration, "DremioClient"]):
def _raw_capabilities(self) -> DestinationCapabilitiesContext:
caps = DestinationCapabilitiesContext()
caps.preferred_loader_file_format = None
caps.supported_loader_file_formats = []
caps.supported_loader_file_formats = ["model"]
caps.preferred_staging_file_format = "parquet"
caps.supported_staging_file_formats = ["jsonl", "parquet", "model"]
caps.supported_staging_file_formats = ["jsonl", "parquet"]
caps.escape_identifier = escape_dremio_identifier
caps.type_mapper = DremioTypeMapper
# all identifiers are case insensitive but are stored as is