f-a-s-by-d-s

aiidalab · Jan 15, 2024 · ca480b1 · ca480b1
1 parent ffe8117
commit ca480b1
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/aiidalab_widgets_base/computational_resources.py b/aiidalab_widgets_base/computational_resources.py
@@ -100,7 +100,7 @@ def __init__(
 
         self._setup_new_code_output = ipw.Output(layout={"width": self._output_width})
 
-        self._default_user_email = orm.User.objects.get_default().email
+        self._default_user_email = orm.User.collection.get_default().email
 
         children = [
             ipw.HBox([self.code_select_dropdown, self.btn_setup_new_code]),
@@ -125,15 +125,7 @@ def __init__(
 
     def _get_codes(self):
         """Query the list of available codes."""
-
-        user = (
-            orm.QueryBuilder()
-            .append(
-                orm.User,
-                filters={"email": self._default_user_email},
-            )
-            .one()[0]
-        )
+        user = orm.User.collection.get(email=self._default_user_email)
 
         filters = (
             {"attributes.input_plugin": self.default_calc_job_plugin}