Skip to content

Commit

Permalink
FIX: update type hints after QRules Fraction implementation
Browse files Browse the repository at this point in the history
  • Loading branch information
redeboer committed Dec 20, 2024
1 parent f5fbe3d commit 64d412b
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions benchmarks/ampform.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,8 @@
from __future__ import annotations

from collections.abc import Sequence
from pprint import pprint
from typing import TYPE_CHECKING
from typing import TYPE_CHECKING, Union

import numpy as np
import pytest
Expand All @@ -17,10 +18,9 @@
from tensorwaves.function.sympy import create_parametrized_function

if TYPE_CHECKING:
from collections.abc import Mapping, Sequence
from collections.abc import Mapping

from ampform.helicity import HelicityModel
from qrules.combinatorics import StateDefinition

from tensorwaves.function import ParametrizedBackendFunction
from tensorwaves.interface import (
Expand All @@ -31,6 +31,8 @@
ParametrizedFunction,
)

StateDefinition = Union[str, tuple[str, Sequence[float]]]


def formulate_amplitude_model(
formalism,
Expand Down

0 comments on commit 64d412b

Please sign in to comment.