From 1ec45b9d4d0276d43757f777c5fadb14f7580402 Mon Sep 17 00:00:00 2001 From: Tibor Reiss Date: Fri, 26 Jul 2024 20:24:05 +0200 Subject: [PATCH] Linting --- integration/test_collection.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/integration/test_collection.py b/integration/test_collection.py index 6ebf81583..fe736a7ae 100644 --- a/integration/test_collection.py +++ b/integration/test_collection.py @@ -64,6 +64,7 @@ def get_numpy_vector(input_list: list) -> Any: try: import numpy as np + return np.array(input_list) except ModuleNotFoundError: return input_list @@ -307,21 +308,28 @@ class TestInsertManyWithTypedDict(TypedDict): [ ( [ - DataObject(properties={"name": "some numpy one"}, vector=get_numpy_vector([1, 2, 3])), + DataObject( + properties={"name": "some numpy one"}, vector=get_numpy_vector([1, 2, 3]) + ), ], False, ), ( [ - DataObject(properties={"name": "some numpy one"}, vector=get_numpy_vector([1, 2, 3])), - DataObject(properties={"name": "some numpy two"}, vector=get_numpy_vector([11, 12, 13])), + DataObject( + properties={"name": "some numpy one"}, vector=get_numpy_vector([1, 2, 3]) + ), + DataObject( + properties={"name": "some numpy two"}, vector=get_numpy_vector([11, 12, 13]) + ), ], False, ), ( [ DataObject( - properties={"name": "some numpy 2d"}, vector=get_numpy_vector([[1, 2, 3], [11, 12, 13]]) + properties={"name": "some numpy 2d"}, + vector=get_numpy_vector([[1, 2, 3], [11, 12, 13]]), ), ], True,