diff --git a/shared/data_store/put_model.py b/shared/data_store/put_model.py index 06be4fc..8ff416e 100644 --- a/shared/data_store/put_model.py +++ b/shared/data_store/put_model.py @@ -18,6 +18,7 @@ def put_model( used as object name.""" assert isinstance(client, Minio) assert isinstance(model, Module) + assert 'MINIO_BUCKET_NAME' in os.environ bucket_name = os.getenv('MINIO_BUCKET_NAME') subfolder = 'models' # save image to buffer