Skip to content

Commit 7d9c491

Browse files
committed
test: check from numpy
1 parent bc9b1c2 commit 7d9c491

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

lantern/tensor.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -193,3 +193,16 @@ class Test(BaseModel):
193193
numbers: Tensor.float().cpu()
194194

195195
Test(numbers=[1, 2, 3])
196+
197+
198+
def test_from_numpy():
199+
from pydantic import BaseModel
200+
201+
class Test(BaseModel):
202+
numbers: Tensor
203+
204+
numbers = np.array([1, 2, 3])
205+
torch_numbers = Test(numbers=numbers).numbers
206+
207+
assert type(torch_numbers) == torch.Tensor
208+
assert np.allclose(torch_numbers.numpy(), numbers)

0 commit comments

Comments
 (0)