Updating BytesIO to bytes

Author: kumar-shivam-ranjan

ads/model/service/oci_datascience_model.py

@@ -878,7 +878,7 @@ def update_custom_metadata_artifact(
     @check_for_model_id(
         msg="Model needs to be saved to the Model Catalog before fetching custom metadata artifact corresponding to that model"
     )
-    def get_custom_metadata_artifact(self, metadata_key_name: str) -> BytesIO:
+    def get_custom_metadata_artifact(self, metadata_key_name: str) -> bytes:
         """Downloads model custom metadata artifact content for specified model metadata key.
 
         Parameters
@@ -887,7 +887,7 @@ def get_custom_metadata_artifact(self, metadata_key_name: str) -> BytesIO:
             The name of the model metadatum in the metadata.
         Returns
         -------
-        BytesIO
+        bytes
                custom metadata artifact content
 
         """
@@ -902,7 +902,7 @@ def get_custom_metadata_artifact(self, metadata_key_name: str) -> BytesIO:
     @check_for_model_id(
         msg="Model needs to be saved to the Model Catalog before fetching defined metadata artifact corresponding to that model"
     )
-    def get_defined_metadata_artifact(self, metadata_key_name: str) -> BytesIO:
+    def get_defined_metadata_artifact(self, metadata_key_name: str) -> bytes:
         """Downloads model defined metadata artifact content for specified model metadata key.
 
         Parameters
@@ -911,7 +911,7 @@ def get_defined_metadata_artifact(self, metadata_key_name: str) -> BytesIO:
             The name of the model metadatum in the metadata.
         Returns
         -------
-        BytesIO
+        bytes
                 Defined metadata artifact content
 
         """