From ddf08eb2738dd4610eebadf84e6d41d2b449ebf1 Mon Sep 17 00:00:00 2001
From: Benoit Chevallier-Mames <benoit.chevalliermames@zama.ai>
Date: Wed, 29 Sep 2021 11:45:30 +0200
Subject: [PATCH] feat: adding management of numpy.square

refs #126
---
 concrete/numpy/tracing.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/concrete/numpy/tracing.py b/concrete/numpy/tracing.py
index 24e19f74b..c62068f7f 100644
--- a/concrete/numpy/tracing.py
+++ b/concrete/numpy/tracing.py
@@ -271,7 +271,7 @@ class NPTracer(BaseTracer):
         numpy.sinh,
         numpy.spacing,
         numpy.sqrt,
-        # numpy.square,
+        numpy.square,
         # numpy.subtract,
         numpy.tan,
         numpy.tanh,