Commit 1ba94f7 1 parent 30603a2 commit 1ba94f7 Copy full SHA for 1ba94f7
File tree 1 file changed +2
-1
lines changed
sdk/python/feast/infra/offline_stores/contrib/spark_offline_store
1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change 30
30
RetrievalMetadata ,
31
31
)
32
32
from feast .infra .registry .base_registry import BaseRegistry
33
- from feast .infra .utils import aws_utils
34
33
from feast .repo_config import FeastConfigBaseModel , RepoConfig
35
34
from feast .saved_dataset import SavedDatasetStorage
36
35
from feast .type_map import spark_schema_to_np_dtypes
@@ -399,6 +398,8 @@ def to_remote_storage(self) -> List[str]:
399
398
400
399
return _list_files_in_folder (output_uri )
401
400
elif self ._config .offline_store .staging_location .startswith ("s3://" ):
401
+ from feast .infra .utils import aws_utils
402
+
402
403
spark_compatible_s3_staging_location = (
403
404
self ._config .offline_store .staging_location .replace (
404
405
"s3://" , "s3a://"
You can’t perform that action at this time.
0 commit comments