@@ -299,7 +299,9 @@ def _get_dataset_input_workunits(self, run: Run) -> Iterable[MetadataWorkUnit]:
299
299
source_type = dataset_input .dataset .source_type
300
300
dataset_tags = {k [1 ]: v [1 ] for k , v in dataset_input .tags }
301
301
dataset = dataset_input .dataset
302
- platform = self ._get_dataset_platform_from_source_type (source_type )
302
+ formatted_platform = self ._get_dataset_platform_from_source_type (
303
+ source_type
304
+ )
303
305
custom_properties = dataset_tags
304
306
formatted_schema = self ._get_dataset_schema (dataset .schema )
305
307
# If the schema is not formatted, pass the schema as a custom property
@@ -308,7 +310,7 @@ def _get_dataset_input_workunits(self, run: Run) -> Iterable[MetadataWorkUnit]:
308
310
# If the dataset is local or code, we create a local dataset reference
309
311
if source_type in ("local" , "code" ):
310
312
local_dataset_reference = Dataset (
311
- platform = platform ,
313
+ platform = formatted_platform ,
312
314
name = dataset .name ,
313
315
schema = formatted_schema ,
314
316
custom_properties = custom_properties ,
@@ -318,7 +320,7 @@ def _get_dataset_input_workunits(self, run: Run) -> Iterable[MetadataWorkUnit]:
318
320
# Otherwise, we create a hosted dataset reference and a hosted dataset
319
321
else :
320
322
hosted_dataset = Dataset (
321
- platform = self . _get_dataset_platform_from_source_type ( source_type ) ,
323
+ platform = formatted_platform ,
322
324
name = dataset .name ,
323
325
schema = formatted_schema ,
324
326
custom_properties = dataset_tags ,
0 commit comments