From 53127549e8e94b2035da388ec6a59fd203f4b3aa Mon Sep 17 00:00:00 2001 From: grayson-helmholz Date: Mon, 11 Nov 2024 11:33:19 +0100 Subject: [PATCH] `Fraction`-literals in `__init__.py` --- src/qrules/__init__.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/qrules/__init__.py b/src/qrules/__init__.py index 9abd1717..cad9070c 100644 --- a/src/qrules/__init__.py +++ b/src/qrules/__init__.py @@ -17,6 +17,7 @@ from __future__ import annotations +from fractions import Fraction from itertools import product from typing import TYPE_CHECKING @@ -73,7 +74,7 @@ def check_reaction_violations( # noqa: C901, PLR0917 mass_conservation_factor: float | None = 3.0, particle_db: ParticleCollection | None = None, max_angular_momentum: int = 1, - max_spin_magnitude: float = 2.0, + max_spin_magnitude: Fraction = Fraction(2, 1), ) -> set[frozenset[str]]: """Determine violated interaction rules for a given particle reaction. @@ -207,7 +208,7 @@ def check_edge_qn_conservation() -> set[frozenset[str]]: InteractionProperties(l_magnitude=l_magnitude, s_magnitude=s_magnitude) for l_magnitude, s_magnitude in product( _int_domain(0, max_angular_momentum), - _halves_domain(0, max_spin_magnitude), + _halves_domain(Fraction(0, 1), max_spin_magnitude), ) ] @@ -272,7 +273,7 @@ def generate_transitions( # noqa: PLR0917 particle_db: ParticleCollection | None = None, mass_conservation_factor: float | None = 3.0, max_angular_momentum: int = 2, - max_spin_magnitude: float = 2.0, + max_spin_magnitude: Fraction = Fraction(2, 1), topology_building: str = "isobar", number_of_threads: int | None = None, ) -> ReactionInfo: