Fixing custom S3 endpoint read fail

llFireHawkll · llFireHawkll · commit d5f1da1561dd · 2022-06-14T02:00:10.000+05:30
Signed-off-by: llFireHawkll &lt;sparsh.dtt@gmail.com&gt;
diff --git 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
