From 7d6738916bbf1d7efcffcc3b9b88a8d7c41626b0 Mon Sep 17 00:00:00 2001
From: Quentin Bourgerie <bourgerie.quentin@gmail.com>
Date: Tue, 17 Aug 2021 15:08:19 +0200
Subject: [PATCH] fix(compiler/pyhton): Update test dot_eint_int

---
 compiler/tests/python/test_round_trip.py | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/compiler/tests/python/test_round_trip.py b/compiler/tests/python/test_round_trip.py
index 25730c755..aab3d9f75 100644
--- a/compiler/tests/python/test_round_trip.py
+++ b/compiler/tests/python/test_round_trip.py
@@ -17,15 +17,13 @@ VALID_INPUTS = [
     }
     """,
     """
-    func @dot_eint_int(%arg0: memref<2x!HLFHE.eint<2>>,
-        %arg1: memref<2xi3>,
-        %arg2: memref<!HLFHE.eint<2>>)
+    func @dot_eint_int(%arg0: tensor<2x!HLFHE.eint<2>>,
+        %arg1: tensor<2xi3>) -> !HLFHE.eint<2>
     {
-        "HLFHE.dot_eint_int"(%arg0, %arg1, %arg2) :
-            (memref<2x!HLFHE.eint<2>>, memref<2xi3>, memref<!HLFHE.eint<2>>) -> ()
-        return
+        %1 = "HLFHE.dot_eint_int"(%arg0, %arg1) :
+            (tensor<2x!HLFHE.eint<2>>, tensor<2xi3>) -> !HLFHE.eint<2>
+        return %1 : !HLFHE.eint<2>
     }
-
     """,
 ]