From 8ded742b29634b9790239903db17c228ea28739a Mon Sep 17 00:00:00 2001 From: Shehtab Zaman Date: Mon, 30 Jan 2023 12:21:52 -0800 Subject: [PATCH] Passing CI tests --- .../unit_tests/test_unit_layer_channelwise_softmax_distconv.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ci_test/unit_tests/test_unit_layer_channelwise_softmax_distconv.py b/ci_test/unit_tests/test_unit_layer_channelwise_softmax_distconv.py index 60f8369ac0d..0db96d05d58 100644 --- a/ci_test/unit_tests/test_unit_layer_channelwise_softmax_distconv.py +++ b/ci_test/unit_tests/test_unit_layer_channelwise_softmax_distconv.py @@ -184,5 +184,5 @@ def construct_data_reader(lbann): # ============================================== # Create test functions that can interact with PyTest -for _test_func in tools.create_tests(setup_experiment, __file__): +for _test_func in tools.create_tests(setup_experiment, __file__, environment=tools.get_distconv_environment()): globals()[_test_func.__name__] = _test_func