fix missing annotations handling in marshmallow (#637)

SwissDataScienceCenter · Apr 27, 2021 · 23e54b0 · 23e54b0
1 parent acc8ded
commit 23e54b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/renku_notebooks/api/schemas.py b/renku_notebooks/api/schemas.py
@@ -130,10 +130,10 @@ class UserPodAnnotations(
     class Meta:
         unknown = INCLUDE
 
-    def get_attribute(self, obj, key, *args, **kwargs):
+    def get_attribute(self, obj, key, default, *args, **kwargs):
         # in marshmallow, any schema key with a dot in it is converted to nested dictionaries
         # in marshmallow, this overrides that behaviour for dumping (serializing)
-        return obj[key]
+        return obj.get(key, default)
 
     @post_load
     def unnest_keys(self, data, **kwargs):