in review_sentiment_flow.py [0:0]
def upload_model_to_gcs(self):
from mozmlops.cloud_storage_api_client import CloudStorageAPIClient
print(f"Uploading model to gcs")
# init client
storage_client = CloudStorageAPIClient(
project_name=GCS_PROJECT_NAME, bucket_name=GCS_BUCKET_NAME
)
storage_client.store(data=self.model_state_dict_bytes, storage_path=MODEL_STORAGE_PATH)
self.next(self.end)