From ab7cf24285e266d2c16b48b033de692e06287491 Mon Sep 17 00:00:00 2001 From: Benoit Chevallier-Mames Date: Wed, 29 Sep 2021 14:07:12 +0200 Subject: [PATCH] fix: pylint. --- tests/numpy/test_tracing.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/numpy/test_tracing.py b/tests/numpy/test_tracing.py index aa47196b0..aa08ab00e 100644 --- a/tests/numpy/test_tracing.py +++ b/tests/numpy/test_tracing.py @@ -259,7 +259,7 @@ def test_tracing_astype( def test_trace_numpy_supported_ufuncs(inputs, expected_output_node): """Function to trace supported numpy ufuncs""" - LIST_OF_UFUNC_WHOSE_OUTPUT_IS_FLOAT64: List[numpy.ufunc] = [ + list_of_ufunc_whose_output_is_float64 = [ # The commented functions are functions which don't work for the moment, often # if not always because they require more than a single argument # numpy.absolute, @@ -365,7 +365,7 @@ def test_trace_numpy_supported_ufuncs(inputs, expected_output_node): assert isinstance(op_graph.output_nodes[0], expected_output_node) assert len(op_graph.output_nodes[0].outputs) == 1 - if function_to_trace_def in LIST_OF_UFUNC_WHOSE_OUTPUT_IS_FLOAT64: + if function_to_trace_def in list_of_ufunc_whose_output_is_float64: assert op_graph.output_nodes[0].outputs[0] == EncryptedScalar(Float(64)) else: assert op_graph.output_nodes[0].outputs[0] in [