💚 Fixes (#1782)

Co-authored-by: Sunny Sun <[email protected]>

lamindb/_parents.py

@@ -153,13 +153,17 @@ def _view_parents(
 
     df_edges = None
     df_edges_parents = _df_edges_from_parents(
-        record=record, field=field, distance=distance
+        record=record, field=field, distance=distance, attr_name=attr_name
     )
     if df_edges_parents is not None:
         df_edges = df_edges_parents
     if with_children:
         df_edges_children = _df_edges_from_parents(
-            record=record, field=field, distance=distance, children=True
+            record=record,
+            field=field,
+            distance=distance,
+            children=True,
+            attr_name=attr_name,
         )
         if df_edges_children is not None:
             if df_edges is not None:
@@ -250,7 +254,11 @@ def _df_edges_from_parents(
     else:
         key = "successors" if children else "predecessors"
     parents = _get_parents(
-        record=record, field=field, distance=distance, children=children
+        record=record,
+        field=field,
+        distance=distance,
+        children=children,
+        attr_name=attr_name,
     )
     all = record.__class__.objects
     records = parents | all.filter(id=record.id)

lamindb/_transform.py

@@ -85,7 +85,7 @@ def latest_run(self) -> Run:
 def view_lineage(self, with_successors: bool = False, distance: int = 5):
     return _view_parents(
         record=self,
-        field=None,
+        field="name",
         with_children=with_successors,
         distance=distance,
         attr_name="predecessors",

lamindb/core/_label_manager.py

@@ -39,6 +39,8 @@ def get_labels_as_dict(self: HasFeatures, links: bool = False):
         "links_feature_set",
         "previous_runs",
         "_feature_values",
+        "_lnschema_core_collection__actions_+",
+        "_actions",
     }
     labels = {}  # type: ignore
     if self.id is None: