Skip to content

Commit 10f9f22

Browse files
committed
fixed comparator with given rtol/atol
1 parent a063e25 commit 10f9f22

File tree

1 file changed

+12
-9
lines changed

1 file changed

+12
-9
lines changed

larray_editor/comparator.py

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -62,15 +62,6 @@ def __init__(self, parent=None, bg_gradient='red-white-blue', rtol=0, atol=0, na
6262

6363
self.arraywidget.btn_layout.addLayout(tolerance_layout)
6464

65-
if rtol > 0 and atol > 0:
66-
raise ValueError("Arguments 'rtol' and 'atol' cannot be used together.")
67-
if rtol > 0:
68-
self.tolerance_combobox.setCurrentText("relative")
69-
self.tolerance_line_edit.setText(str(rtol))
70-
if atol > 0:
71-
self.tolerance_combobox.setCurrentText("absolute")
72-
self.tolerance_line_edit.setText(str(atol))
73-
7465
self.nans_equal = nans_equal
7566

7667
# add local arraywidget to layout
@@ -83,6 +74,15 @@ def __init__(self, parent=None, bg_gradient='red-white-blue', rtol=0, atol=0, na
8374
self.bg_value = None
8475
self.stack_axis = None
8576

77+
if rtol > 0 and atol > 0:
78+
raise ValueError("Arguments 'rtol' and 'atol' cannot be used together.")
79+
if rtol > 0:
80+
self.tolerance_combobox.setCurrentText("relative")
81+
self.tolerance_line_edit.setText(str(rtol))
82+
if atol > 0:
83+
self.tolerance_combobox.setCurrentText("absolute")
84+
self.tolerance_line_edit.setText(str(atol))
85+
8686
# override keyPressEvent to prevent pressing Enter after changing the tolerance value
8787
# in associated QLineEdit to close the parent dialog box
8888
def keyPressEvent(self, event):
@@ -111,6 +111,9 @@ def set_data(self, arrays, stack_axis):
111111
self.update_isequal()
112112

113113
def update_isequal(self):
114+
if self.array is None:
115+
return
116+
114117
try:
115118
tol_str = self.tolerance_line_edit.text()
116119
tol = ast.literal_eval(tol_str) if tol_str else 0

0 commit comments

Comments
 (0)