diff --git a/eralchemy/models.py b/eralchemy/models.py index b54b36e9c64950d92454cb57bdf38d797cf9f2f3..fc906801b9ad75e065eeaf8edcffe84f918d3a10 100644 --- a/eralchemy/models.py +++ b/eralchemy/models.py @@ -204,12 +204,8 @@ class Relation(Drawable): return '{} "{}" -- "{}" {}'.format(*normalized) def to_mermaid_er(self) -> str: - left = Relation.cardinalities_crowfoot.get( - self.left_cardinality, - ) - right = Relation.cardinalities_crowfoot.get( - self.right_cardinality, - ) + left = Relation.cardinalities_crowfoot.get(self.left_cardinality, self.left_cardinality) + right = Relation.cardinalities_crowfoot.get(self.right_cardinality, self.right_cardinality) left_col = sanitize_mermaid(self.left_table, is_er=True) right_col = sanitize_mermaid(self.right_table, is_er=True)