From 9ffe9b667ac9e44f2ed3f2a9120d8c6c74ba80b6 Mon Sep 17 00:00:00 2001 From: Arthur Meyre Date: Thu, 2 Sep 2021 14:07:30 +0200 Subject: [PATCH] refacto: fix missed renaming after bad merge --- hdk/numpy/tracing.py | 2 +- tests/numpy/test_tracing.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/hdk/numpy/tracing.py b/hdk/numpy/tracing.py index d63b13b16..d5247dc38 100644 --- a/hdk/numpy/tracing.py +++ b/hdk/numpy/tracing.py @@ -42,7 +42,7 @@ class NPTracer(BaseTracer): tracing_func = self.get_tracing_func_for_np_function(ufunc) assert ( len(kwargs) == 0 - ), f"hnumpy does not support **kwargs currently for numpy ufuncs, ufunc: {ufunc}" + ), f"**kwargs are currently not supported for numpy ufuncs, ufunc: {ufunc}" return tracing_func(*input_tracers, **kwargs) raise NotImplementedError("Only __call__ method is supported currently") diff --git a/tests/numpy/test_tracing.py b/tests/numpy/test_tracing.py index fa03267d8..521c1bec6 100644 --- a/tests/numpy/test_tracing.py +++ b/tests/numpy/test_tracing.py @@ -280,7 +280,7 @@ def test_trace_numpy_supported_ufuncs(inputs, expected_output_node, expected_out assert op_graph.output_nodes[0].outputs[0] == expected_output_value -def test_trace_hnumpy_ufuncs_not_supported(): +def test_trace_numpy_ufuncs_not_supported(): """Testing a failure case of trace_numpy_function""" inputs = {"x": EncryptedScalar(Integer(128, is_signed=True))}