Skip to content

Commit

Permalink
use math instead of np.math
Browse files Browse the repository at this point in the history
  • Loading branch information
JoschD committed Jan 11, 2024
1 parent d4839e1 commit 8a7bd85
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion irnl_rdt_correction/equation_system.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
"""
import logging
import math
from typing import Dict, List, Sequence, Set, Tuple

import numpy as np
Expand Down Expand Up @@ -354,7 +355,7 @@ def get_elements_integral(rdt: RDT, ip: int, optics: Optics, feeddown: int) -> f
iksl_err = 1j*errors_df.loc[elements, f"K{n_mad:d}SL"]

k_sum += ((kl_opt + kl_err + iksl_opt + iksl_err) *
(dx_idy**q) / np.math.factorial(q))
(dx_idy**q) / math.factorial(q))

# note the minus sign before the sum!
integral += -sum(np.real(i_pow(lm) * k_sum.to_numpy()) * (side_sign * betax * betay).to_numpy())
Expand Down

0 comments on commit 8a7bd85

Please sign in to comment.