|
64 | 64 | def get_numpy_vector(input_list: list) -> Any: |
65 | 65 | try: |
66 | 66 | import numpy as np |
| 67 | + |
67 | 68 | return np.array(input_list) |
68 | 69 | except ModuleNotFoundError: |
69 | 70 | return input_list |
@@ -307,21 +308,28 @@ class TestInsertManyWithTypedDict(TypedDict): |
307 | 308 | [ |
308 | 309 | ( |
309 | 310 | [ |
310 | | - DataObject(properties={"name": "some numpy one"}, vector=get_numpy_vector([1, 2, 3])), |
| 311 | + DataObject( |
| 312 | + properties={"name": "some numpy one"}, vector=get_numpy_vector([1, 2, 3]) |
| 313 | + ), |
311 | 314 | ], |
312 | 315 | False, |
313 | 316 | ), |
314 | 317 | ( |
315 | 318 | [ |
316 | | - DataObject(properties={"name": "some numpy one"}, vector=get_numpy_vector([1, 2, 3])), |
317 | | - DataObject(properties={"name": "some numpy two"}, vector=get_numpy_vector([11, 12, 13])), |
| 319 | + DataObject( |
| 320 | + properties={"name": "some numpy one"}, vector=get_numpy_vector([1, 2, 3]) |
| 321 | + ), |
| 322 | + DataObject( |
| 323 | + properties={"name": "some numpy two"}, vector=get_numpy_vector([11, 12, 13]) |
| 324 | + ), |
318 | 325 | ], |
319 | 326 | False, |
320 | 327 | ), |
321 | 328 | ( |
322 | 329 | [ |
323 | 330 | DataObject( |
324 | | - properties={"name": "some numpy 2d"}, vector=get_numpy_vector([[1, 2, 3], [11, 12, 13]]) |
| 331 | + properties={"name": "some numpy 2d"}, |
| 332 | + vector=get_numpy_vector([[1, 2, 3], [11, 12, 13]]), |
325 | 333 | ), |
326 | 334 | ], |
327 | 335 | True, |
|
0 commit comments