only check if data component is serializable

materialsproject · Aug 17, 2022 · 9999390 · 9999390
1 parent 96e9772
commit 9999390
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/mpcontribs-client/mpcontribs/client/__init__.py b/mpcontribs-client/mpcontribs/client/__init__.py
@@ -1350,9 +1350,10 @@ def update_contributions(
         if not valid:
             return {"error": error}
 
-        serializable, error = self._is_serializable_dict(data)
-        if not serializable:
-            return {"error": error}
+        if "data" in data:
+            serializable, error = self._is_serializable_dict(data["data"])
+            if not serializable:
+                return {"error": error}
 
         query = query or {}
 
@@ -1580,9 +1581,10 @@ def submit_contributions(
         fields.remove("needs_build")  # internal field
 
         for contrib in tqdm(contributions, desc="Prepare"):
-            serializable, error = self._is_serializable_dict(contrib)
-            if not serializable:
-                raise ValueError(error)
+            if "data" in contrib:
+                serializable, error = self._is_serializable_dict(contrib["data"])
+                if not serializable:
+                    raise ValueError(error)
 
             update = "id" in contrib
             project_name = id2project[contrib["id"]] if update else contrib["project"]