change cache reset to work on User.obects only

implement specific cache reset method for User.objects
aiidateam · Oct 27, 2019 · 61d696e · 61d696e
1 parent 1a636d0
commit 61d696e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 9 deletions.
diff --git a/aiida/orm/entities.py b/aiida/orm/entities.py
@@ -163,15 +163,6 @@ def all(self):
         """
         return [_[0] for _ in self.query().all()]
 
-    def reset(self):
-        """
-        Remove lazily constructed collection instance.
-
-        Removes lazily constructed collection instance from LazyStore (constructed again upon next request).
-        Useful to clear any internal caches of the collection.
-        """
-        self._COLLECTIONS.pop((self.entity_type, self.backend))
-
 
 class Entity(object):
     """An AiiDA entity"""

diff --git a/aiida/orm/users.py b/aiida/orm/users.py
@@ -72,6 +72,12 @@ def get_default(self):
 
             return self._default_user
 
+        def reset(self):
+            """
+            Reset internal caches (default user).
+            """
+            self._default_user = self.UNDEFINED
+
     REQUIRED_FIELDS = ['first_name', 'last_name', 'institution']
 
     def __init__(self, email, first_name='', last_name='', institution='', backend=None):