Merge pull request FedML-AI#1707 from FedML-AI/alexleung/dev_branch

[CoreEngine] fixed the issue when merging storage uploading.
jrpizani · Dec 5, 2023 · f5452aa · f5452aa
2 parents a7c67bf + 8d3b290
commit f5452aa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/fedml/api/modules/storage.py b/python/fedml/api/modules/storage.py
@@ -6,7 +6,7 @@
 from fedml.core.mlops.mlops_configs import Configs, MLOpsConfigs
 
 
-def upload(data_path, api_key, name, show_progress, out_progress_to_err, progress_desc) -> str | None:
+def upload(data_path, api_key, name, show_progress, out_progress_to_err, progress_desc) -> str:
     authenticate(api_key)
     configs = MLOpsConfigs.fetch_remote_storage_configs()
     r2_config = configs[Configs.R2_CONFIG]
@@ -39,7 +39,7 @@ def upload(data_path, api_key, name, show_progress, out_progress_to_err, progres
     return file_uploaded_url
 
 
-def download(data_name, api_key) -> str | None:
+def download(data_name, api_key) -> str:
     authenticate(api_key)
     configs = MLOpsConfigs.fetch_remote_storage_configs()
     r2_config = configs[Configs.R2_CONFIG]