make EMR jar uploader work the same as k8s one

Signed-off-by: Oleg Avdeev <[email protected]>
feast-dev · Jan 23, 2021 · 80d379e · 80d379e
1 parent 775d673
commit 80d379e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sdk/python/feast/pyspark/launchers/aws/emr_utils.py b/sdk/python/feast/pyspark/launchers/aws/emr_utils.py
@@ -81,7 +81,11 @@ def _random_string(length) -> str:
 
 
 def _upload_jar(jar_s3_prefix: str, jar_path: str) -> str:
-    if jar_path.startswith("https://"):
+    if (
+        jar_path.startswith("s3://")
+        or jar_path.startswith("s3a://")
+        or jar_path.startswith("https://")
+    ):
         return jar_path
     with open(jar_path, "rb") as f:
         uri = urlparse(os.path.join(jar_s3_prefix, os.path.basename(jar_path)))