Skip to content

Commit

Permalink
removed __render_as_fraction altogether
Browse files Browse the repository at this point in the history
  • Loading branch information
grayson-helmholz committed Nov 25, 2024
1 parent c061be1 commit 3e2ba1a
Showing 1 changed file with 1 addition and 13 deletions.
14 changes: 1 addition & 13 deletions src/qrules/io/_dot.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,6 @@
from fractions import Fraction
from functools import singledispatch
from inspect import isfunction
from numbers import Number
from typing import TYPE_CHECKING, Any, cast

import attrs
Expand Down Expand Up @@ -317,22 +316,11 @@ def _(obj: dict) -> str:
key_repr = key
if value != 0 or any(s in key_repr for s in ["magnitude", "projection"]):
pm = not any(s in key_repr for s in ["pid", "mass", "width", "magnitude"])
value_repr = __render_as_fraction(value, pm)
value_repr = _render_fraction(value, pm)
lines.append(f"{key_repr} = {value_repr}")
return "\n".join(lines)


def __render_as_fraction(value: Any, plusminus: bool) -> str:
plusminus &= isinstance(value, Number) and bool(value)
if isinstance(value, float):
return _render_fraction(Fraction(value), plusminus)
if isinstance(value, Fraction):
return _render_fraction(value, plusminus)
if plusminus:
return f"{value:+}"
return str(value)


def _render_fraction(fraction: Fraction, plusminus: bool = False) -> str:
if plusminus:
if fraction.denominator == 1:
Expand Down

0 comments on commit 3e2ba1a

Please sign in to comment.