diff --git a/sdk/python/feast/infra/offline_stores/file_source.py b/sdk/python/feast/infra/offline_stores/file_source.py index 5c2a521ac0..85baa64703 100644 --- a/sdk/python/feast/infra/offline_stores/file_source.py +++ b/sdk/python/feast/infra/offline_stores/file_source.py @@ -179,9 +179,15 @@ def get_table_column_names_and_types( filesystem, path = FileSource.create_filesystem_and_path( self.path, self.file_options.s3_endpoint_override ) - schema = ParquetDataset( - path if filesystem is None else filesystem.open_input_file(path) - ).schema.to_arrow_schema() + # Adding support for different file format path + # based on S3 filesystem + if filesystem is None: + schema = ParquetDataset(path).schema.to_arrow_schema() + else: + schema = ParquetDataset( + filesystem.open_input_file(path), filesystem=filesystem + ).schema + return zip(schema.names, map(str, schema.types)) @staticmethod