Skip to content

Commit 265389c

Browse files
committed
fix: handle extra arguments to validate
1 parent 645eb12 commit 265389c

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

lantern/numpy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ def __get_validators__(cls):
1010
yield cls.validate
1111

1212
@classmethod
13-
def validate(cls, data) -> np.ndarray:
13+
def validate(cls, data, config=None, field=None) -> np.ndarray:
1414
if isinstance(data, cls):
1515
return data.view(np.ndarray)
1616
elif isinstance(data, np.ndarray):

lantern/tensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def __get_validators__(cls):
2020
yield cls.validate
2121

2222
@classmethod
23-
def validate(cls, data) -> torch.Tensor:
23+
def validate(cls, data, config=None, field=None) -> torch.Tensor:
2424
if isinstance(data, cls):
2525
return torch.tensor(data)
2626
elif isinstance(data, torch.Tensor):

0 commit comments

Comments
 (0)