Skip to content

Commit ddcfd7d

Browse files
committed
STY: fix black issues
1 parent acd72e7 commit ddcfd7d

File tree

3 files changed

+5
-6
lines changed

3 files changed

+5
-6
lines changed

refellips/dataSE.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
A basic representation of a 1D dataset
55
"""
66

7+
import copy
78
import numpy as np
89
import pandas as pd
910
from refnx._lib import possibly_open_file

refellips/dispersion.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -562,9 +562,7 @@ def epsilon(self, energy):
562562
+ Eg**2 * C**2
563563
- Ei**2 * (Ei**2 + 3 * Eg**2)
564564
)
565-
a_atan = (energies**2 - Ei**2) * (
566-
Ei**2 + Eg**2
567-
) + Eg**2 * C**2
565+
a_atan = (energies**2 - Ei**2) * (Ei**2 + Eg**2) + Eg**2 * C**2
568566
alpha = np.sqrt(4 * Ei**2 - C**2)
569567
gamma = np.sqrt(Ei**2 - C**2 / 2)
570568
zeta4 = (energies**2 - gamma**2) ** 2 + 0.25 * alpha**2 * C**2
@@ -579,8 +577,7 @@ def epsilon(self, energy):
579577
/ alpha
580578
/ Ei
581579
* np.log(
582-
(Ei**2 + Eg**2 + alpha * Eg)
583-
/ (Ei**2 + Eg**2 - alpha * Eg)
580+
(Ei**2 + Eg**2 + alpha * Eg) / (Ei**2 + Eg**2 - alpha * Eg)
584581
)
585582
)
586583
e1 -= (

refellips/objectiveSE.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -379,8 +379,9 @@ def logl(self, pvals=None):
379379
# here just set it to unity
380380
y_err = 1
381381
if self.lnsigma is not None:
382+
_model = np.r_[psi, delta]
382383
var_y = (
383-
y_err * y_err + np.exp(2 * float(self.lnsigma)) * model * model
384+
y_err * y_err + np.exp(2 * float(self.lnsigma)) * _model * _model
384385
)
385386
else:
386387
var_y = y_err**2

0 commit comments

Comments
 (0)