linting

saturncloud · Oct 8, 2024 · 7f2dde4 · 7f2dde4
1 parent f2fe913
commit 7f2dde4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/saturn_client/core.py b/saturn_client/core.py
@@ -363,6 +363,11 @@ def delete_shared_folder(self, shared_folder_id: str) -> Dict:
         response = self.session.delete(url)
         return response.json()
 
+    def set_preferred_org(self, user_id: str, org_id: str) -> Dict:
+        url = urljoin(self.url, "api/user/preferences")
+        response = self.session.post(url, json={"user_id": user_id, "default_org_id": org_id})
+        return response.json()
+
     def get_size(self, size: str) -> Dict:
         sizes = self.list_options(ServerOptionTypes.SIZES)
         pruned = [x for x in sizes if x["name"] == size]