replaced other occurences of 'Any' with qn-types

src/qrules/conservation_rules.py

@@ -55,8 +55,12 @@
 from attrs.converters import optional
 
 from qrules.quantum_numbers import EdgeQuantumNumbers as EdgeQN
+from qrules.quantum_numbers import (
+    EdgeQuantumNumberTypes,
+    NodeQuantumNumberTypes,
+    arange,
+)
 from qrules.quantum_numbers import NodeQuantumNumbers as NodeQN
-from qrules.quantum_numbers import arange
 
 
 def _is_boson(spin_magnitude: float) -> bool:
@@ -75,15 +79,18 @@ def __call__(self, qns: Any, /) -> bool: ...
 
 class EdgeQNConservationRule(Protocol):
     def __call__(
-        self, ingoing_edge_qns: list[Any], outgoing_edge_qns: list[Any], /
+        self,
+        ingoing_edge_qns: list[EdgeQuantumNumberTypes],
+        outgoing_edge_qns: list[EdgeQuantumNumberTypes],
+        /,
     ) -> bool: ...
 
 
 class ConservationRule(Protocol):
     def __call__(
         self,
-        ingoing_edge_qns: list[Any],
-        outgoing_edge_qns: list[Any],
+        ingoing_edge_qns: list[EdgeQuantumNumberTypes],
+        outgoing_edge_qns: list[EdgeQuantumNumberTypes],
         node_qns: Any,
         /,
     ) -> bool: ...
@@ -93,7 +100,7 @@ def __call__(
 # __call__ method in a concrete version of the generic are still containing the
 # TypeVar types. See https://github.com/python/typing/issues/762
 def additive_quantum_number_rule(
-    quantum_number: type,
+    quantum_number: Union[EdgeQuantumNumberTypes, NodeQuantumNumberTypes],
 ) -> Callable[[Any], EdgeQNConservationRule]:
     r"""Class decorator for creating an additive conservation rule.