fix(frontend-python): support concatenating signed and unsigned tensors together

This commit is contained in:
Umut
2023-06-20 14:19:44 +02:00
parent 9363c40753
commit f2929df39d
2 changed files with 8 additions and 1 deletions

View File

@@ -647,7 +647,7 @@ class Context:
)
)
x = self.encrypt(encrypted_type, x)
sanitized_xs.append(x)
sanitized_xs.append(self.to_signedness(x, of=resulting_type))
if axis is None:
return self.operation(

View File

@@ -171,6 +171,13 @@ from concrete import fhe
"y": {"shape": (5, 2, 3)},
},
),
pytest.param(
lambda x, y: np.concatenate((x, y)),
{
"x": {"range": [0, 10], "shape": (4, 2)},
"y": {"range": [-10, 10], "shape": (3, 2)},
},
),
],
)
def test_concatenate(function, parameters, helpers):