fix: printing of dataset, instantiated by ctx.get_resource(rid)

emdgroup · Sep 13, 2024 · 4b0cc0f · 4b0cc0f
1 parent e8e07df
commit 4b0cc0f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/libs/foundry-dev-tools/src/foundry_dev_tools/resources/dataset.py b/libs/foundry-dev-tools/src/foundry_dev_tools/resources/dataset.py
@@ -797,7 +797,9 @@ def sync(self) -> Self:
 
     def _get_repr_dict(self) -> dict:
         d = super()._get_repr_dict()
-        d["branch"] = d["branch"]["id"]
+        if not hasattr(self, "branch"):
+            self.branch = self._context.catalog.api_get_branch(self.rid, "master").json()
+        d["branch"] = self.branch["id"]
         return d