Merge pull request #241 from jrzaurin/fix-broken-tests

limit the torch version to <2.6 for now

Author: jrzaurin

pytorch_widedeep/losses.py

@@ -896,7 +896,7 @@ def _compute_cont_loss(self, x_cont_and_cont_) -> Tensor:
 
     @staticmethod
     def _get_device(
-        x_and_x_: Union[List[Tuple[Tensor, Tensor]], Tuple[Tensor, Tensor]]
+        x_and_x_: Union[List[Tuple[Tensor, Tensor]], Tuple[Tensor, Tensor]],
     ):
         if isinstance(x_and_x_, tuple):
             device = x_and_x_[0].device

requirements.txt

@@ -6,7 +6,7 @@ gensim
 spacy
 opencv-contrib-python>=4.9.0.80
 tqdm
-torch >= 2.0.0
+torch >= 2.0.0, <2.6.0
 torchvision >= 0.15.0
 einops
 wrapt