Guard against immutable ID not being returned from create collection (#…

…35)
datalab-org · Oct 12, 2024 · 382d9b4 · 382d9b4
1 parent 7376b54
commit 382d9b4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/datalab_api/__init__.py b/src/datalab_api/__init__.py
@@ -131,7 +131,8 @@ def create_item(
             try:
                 collection_immutable_id = self.get_collection(collection_id)["immutable_id"]
             except RuntimeError:
-                collection_immutable_id = self.create_collection(collection_id)["immutable_id"]
+                self.create_collection(collection_id)
+                collection_immutable_id = self.get_collection(collection_id)["immutable_id"]
             new_item["collections"] = new_item.get("collections", [])
             new_item["collections"].append({"immutable_id": collection_immutable_id})