def upload_model_to_gcs()

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)