From 6f7851041fc513d13b9f89d96896e1dd12719cf5 Mon Sep 17 00:00:00 2001
From: Shyam D <shyamd@lbl.gov>
Date: Tue, 29 Sep 2020 10:40:06 -0700
Subject: [PATCH] convert tags to sandboxes in builder

---
 emmet-builders/emmet/builders/vasp/materials.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/emmet-builders/emmet/builders/vasp/materials.py b/emmet-builders/emmet/builders/vasp/materials.py
index 2cde2643af..0560f04c2d 100644
--- a/emmet-builders/emmet/builders/vasp/materials.py
+++ b/emmet-builders/emmet/builders/vasp/materials.py
@@ -51,6 +51,7 @@ def __init__(
         task_validation: Optional[Store] = None,
         query: Optional[Dict] = None,
         allowed_task_types: Optional[List[str]] = None,
+        tags_to_sandoxes: Optional[Dict[str, List[str]]] = None,
         symprec: float = SETTINGS.SYMPREC,
         ltol: float = SETTINGS.LTOL,
         stol: float = SETTINGS.STOL,
@@ -63,6 +64,7 @@ def __init__(
             materials: Store of materials documents to generate
             query: dictionary to limit tasks to be analyzed
             allowed_task_types: list of task_types that can be processed
+            tags_to_sandboxes: dictionary mapping sandboxes to a list of tags
             symprec: tolerance for SPGLib spacegroup finding
             ltol: StructureMatcher tuning parameter for matching tasks to materials
             stol: StructureMatcher tuning parameter for matching tasks to materials
@@ -75,6 +77,7 @@ def __init__(
         self.allowed_task_types = {TaskType(t) for t in allowed_task_types} or set(
             TaskType
         )
+        self.tags_to_sandoxes = tags_to_sandoxes or SETTINGS.tags_to_sandboxes
         self.query = query if query else {}
         self.symprec = symprec
         self.ltol = ltol
@@ -170,7 +173,7 @@ def get_items(self) -> Iterator[List[Dict]]:
             "output.parameters",
             "orig_inputs",
             "input.structure",
-            "sandboxes",
+            "tags",
         ]
 
         for formula in to_process_forms:
@@ -185,6 +188,14 @@ def get_items(self) -> Iterator[List[Dict]]:
                 else:
                     t["is_valid"] = True
 
+                t["sandboxes"] = [
+                    sandbox
+                    for sandbox in self.tags_to_sandoxes
+                    if any(
+                        tag in t["tags"] for tag in set(self.tags_to_sandoxes[sandbox])
+                    )
+                ]
+
             yield tasks
 
     def process_item(self, tasks: List[Dict]) -> List[Dict]: