api: update init_columns() for sampling

materialsproject · Oct 3, 2022 · 041359c · 041359c
1 parent 315eb4c
commit 041359c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/mpcontribs-api/mpcontribs/api/projects/document.py b/mpcontribs-api/mpcontribs/api/projects/document.py
@@ -237,10 +237,10 @@ def post_save(cls, sender, document, **kwargs):
                     flat = flatten(remap(merged, visit=visit, enter=enter), reducer="dot")
 
                     for k, v in flat.items():
-                        path = f"data.{k}"
-                        columns[path] = Column(path=path)
-                        if v is not None:
-                            columns[path].unit = v
+                        if k.startswith("data."):
+                            columns[k] = Column(path=k)
+                            if v is not None:
+                                columns[k].unit = v
 
                 # start pipeline for stats: match project
                 pipeline = [{"$match": {"project": document.id}}]