fix: handle extra arguments to validate

nextml-code · Nov 28, 2023 · 265389c · 265389c
1 parent 645eb12
commit 265389c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lantern/numpy.py b/lantern/numpy.py
@@ -10,7 +10,7 @@ def __get_validators__(cls):
         yield cls.validate
 
     @classmethod
-    def validate(cls, data) -> np.ndarray:
+    def validate(cls, data, config=None, field=None) -> np.ndarray:
         if isinstance(data, cls):
             return data.view(np.ndarray)
         elif isinstance(data, np.ndarray):

diff --git a/lantern/tensor.py b/lantern/tensor.py
@@ -20,7 +20,7 @@ def __get_validators__(cls):
         yield cls.validate
 
     @classmethod
-    def validate(cls, data) -> torch.Tensor:
+    def validate(cls, data, config=None, field=None) -> torch.Tensor:
         if isinstance(data, cls):
             return torch.tensor(data)
         elif isinstance(data, torch.Tensor):