From a0726857a0e4833f67780eb7ea3e3ed15e5a4d0c Mon Sep 17 00:00:00 2001
From: brian <schnitzelen@gmail.com>
Date: Sat, 19 Oct 2024 19:45:16 +0000
Subject: [PATCH] ensure env var set

---
 shared/data_store/put_model.py | 1 +
 1 file changed, 1 insertion(+)

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