adding raise e

Author: Ziqun Ye

ads/opctl/backend/local.py

@@ -650,7 +650,6 @@ def predict(self) -> None:
         model_folder = os.path.expanduser(self.config["execution"].get("model_save_folder", DEFAULT_MODEL_FOLDER))
         artifact_directory = artifact_directory or os.path.join(model_folder, str(ocid))
         if ocid and (not os.path.exists(artifact_directory) or len(os.listdir(artifact_directory)) == 0):
-            
             region = self.config["execution"].get("region", None)
             bucket_uri = self.config["execution"].get("bucket_uri", None)
             timeout = self.config["execution"].get("timeout", None)
@@ -718,8 +717,10 @@ def _download_model(self, ocid, artifact_directory, region, bucket_uri, timeout)
             timeout=timeout,
             bucket_uri=bucket_uri,
             )
-        except:
+            
+        except Exception as e :
             shutil.rmtree(artifact_directory, ignore_errors=True)
+            raise e
 
     def _get_conda_info_from_catalog(self, ocid):
         response = self.client.get_model(ocid)

ads/opctl/conda/cmds.py

@@ -310,6 +310,8 @@ def _install(
     None
     """
     ns, bucket, path, slug = parse_conda_uri(conda_uri)
+    if bucket == "service-conda-packs":
+        raise ValueError("Download service conda pack is not allowed. Only custom conda pack can be downloaded to local machine. You need to publish it to your own bucket first.")
     os.makedirs(conda_pack_folder, exist_ok=True)
     pack_path = os.path.join(os.path.expanduser(conda_pack_folder), slug + ".tar.gz")
     pack_folder_path = os.path.join(os.path.expanduser(conda_pack_folder), slug)