diff --git a/src/LineageTree/lineageTree.py b/src/LineageTree/lineageTree.py index ec187e8..82cb1f3 100644 --- a/src/LineageTree/lineageTree.py +++ b/src/LineageTree/lineageTree.py @@ -3486,7 +3486,9 @@ def __init__( self.pos = { node: np.array(position) for node, position in pos.items() } - + if "labels" in kwargs: + self._labels = kwargs["labels"] + kwargs.pop("labels") if time is None: if starting_time is None: starting_time = 0