Skip to content

Commit f479e00

Browse files
committed
Change the check metric test from using 1e-8 to
np.finfo(np.float32).eps (which is actually ~1.192e-07). This should resolve errors with the test failing.
1 parent 794b266 commit f479e00

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ci_test/unit_tests/test_unit_metric_python.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -109,8 +109,8 @@ def test_metric_with_callback():
109109
tester.extra_callbacks.extend([
110110
lbann.CallbackDumpWeights(directory='.', epoch_interval=1),
111111
lbann.CallbackCheckMetric(metric='pymetric',
112-
lower_bound=w[0] + 1 - 1e-8,
113-
upper_bound=w[0] + 1 + 1e-8,
112+
lower_bound=w[0] + 1 - np.finfo(np.float32).eps,
113+
upper_bound=w[0] + 1 + np.finfo(np.float32).eps,
114114
error_on_failure=True,
115115
execution_modes='test'),
116116
])

0 commit comments

Comments
 (0)