DAL: Remember to update Record attrs on Record.update()

AgnostiqHQ · Nov 1, 2023 · a67e38e · a67e38e
1 parent 9a896a6
commit a67e38e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/covalent_dispatcher/_dal/controller.py b/covalent_dispatcher/_dal/controller.py
@@ -163,6 +163,8 @@ def update(self, session: Session, *, values: dict):
             equality_filters={"id": self.primary_key},
             membership_filters={},
         )
+        for k, v in values.items():
+            self._attrs[k] = v
 
     def incr(self, session: Session, *, increments: dict):
         """Increment the fields of the corresponding record."""