Clean diff for #342

meedan · Oct 25, 2023 · bd14839 · bd14839
1 parent e3f2bd3
commit bd14839
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/app/main/lib/shared_models/shared_model.py b/app/main/lib/shared_models/shared_model.py
@@ -43,7 +43,9 @@ def get_servers():
   @staticmethod
   def get_client(model_key, options={}):
     r = redis.Redis(host=app.config['REDIS_HOST'], port=app.config['REDIS_PORT'], db=app.config['REDIS_DATABASE'])
-    model_class = json.loads(r.get('SharedModel:%s' % model_key).decode('utf-8'))['model_class']
+    model_info = r.get("SharedModel:%s" % model_key)
+    assert model_info is not None, f"Unable locate model info for key {model_key}"
+    model_class = json.loads(model_info.decode('utf-8'))['model_class']
     class_ = SharedModel.import_model_class(model_class)
     return class_(model_key, options)