|
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