Skip more torch

Author: chezou

tests/models/nn/test_item_net.py

@@ -12,27 +12,38 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
+import sys
 import typing as tp
 
 import numpy as np
 import pandas as pd
 import pytest
-import torch
-from pytorch_lightning import seed_everything
+
+try:
+    import torch
+    from pytorch_lightning import seed_everything
+except ImportError:
+    pass
 
 from rectools.columns import Columns
 from rectools.dataset import Dataset
 from rectools.dataset.dataset import DatasetSchema, EntitySchema
-from rectools.models.nn.item_net import (
-    CatFeaturesItemNet,
-    IdEmbeddingsItemNet,
-    ItemNetBase,
-    ItemNetConstructorBase,
-    SumOfEmbeddingsConstructor,
-)
+
+try:
+    from rectools.models.nn.item_net import (
+        CatFeaturesItemNet,
+        IdEmbeddingsItemNet,
+        ItemNetBase,
+        ItemNetConstructorBase,
+        SumOfEmbeddingsConstructor,
+    )
+except ImportError:
+    pass
 
 from ..data import DATASET, INTERACTIONS
 
+pytestmark = pytest.mark.skipif(sys.version_info >= (3, 13), reason="`torch` is not compatible with Python >= 3.13")
+
 
 class TestIdEmbeddingsItemNet:
     def setup_method(self) -> None:

tests/models/nn/transformers/test_base.py

@@ -13,26 +13,38 @@
 #  limitations under the License.
 
 import os
+import sys
 import typing as tp
 from tempfile import NamedTemporaryFile
 
 import pandas as pd
 import pytest
-import torch
 from pytest import FixtureRequest
-from pytorch_lightning import Trainer, seed_everything
-from pytorch_lightning.loggers import CSVLogger
+
+try:
+    import torch
+    from pytorch_lightning import Trainer, seed_everything
+    from pytorch_lightning.loggers import CSVLogger
+
+except ImportError:
+    pass
 
 from rectools import Columns
 from rectools.dataset import Dataset
-from rectools.models import BERT4RecModel, SASRecModel, load_model
-from rectools.models.nn.item_net import CatFeaturesItemNet, IdEmbeddingsItemNet
-from rectools.models.nn.transformers.base import TransformerModelBase
+
+try:
+    from rectools.models import BERT4RecModel, SASRecModel, load_model
+    from rectools.models.nn.item_net import CatFeaturesItemNet, IdEmbeddingsItemNet
+    from rectools.models.nn.transformers.base import TransformerModelBase
+except ImportError:
+    pass
 from tests.models.data import INTERACTIONS
 from tests.models.utils import assert_save_load_do_not_change_model
 
 from .utils import custom_trainer, custom_trainer_ckpt, custom_trainer_multiple_ckpt, leave_one_out_mask
 
+pytestmark = pytest.mark.skipif(sys.version_info >= (3, 13), reason="`torch` is not compatible with Python >= 3.13")
+
 
 class TestTransformerModelBase:
     def setup_method(self) -> None:

tests/models/nn/transformers/test_bert4rec.py

@@ -12,27 +12,36 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
+import sys
 import typing as tp
 from functools import partial
 
 import numpy as np
 import pandas as pd
 import pytest
-import torch
-from pytorch_lightning import Trainer, seed_everything
+
+try:
+    import torch
+    from pytorch_lightning import Trainer, seed_everything
+except ImportError:
+    pass
 
 from rectools import ExternalIds
 from rectools.columns import Columns
 from rectools.dataset import Dataset
-from rectools.models import BERT4RecModel
-from rectools.models.nn.item_net import IdEmbeddingsItemNet, SumOfEmbeddingsConstructor
-from rectools.models.nn.transformers.base import (
-    LearnableInversePositionalEncoding,
-    PreLNTransformerLayers,
-    TrainerCallable,
-    TransformerLightningModule,
-)
-from rectools.models.nn.transformers.bert4rec import MASKING_VALUE, BERT4RecDataPreparator, ValMaskCallable
+
+try:
+    from rectools.models import BERT4RecModel
+    from rectools.models.nn.item_net import IdEmbeddingsItemNet, SumOfEmbeddingsConstructor
+    from rectools.models.nn.transformers.base import (
+        LearnableInversePositionalEncoding,
+        PreLNTransformerLayers,
+        TrainerCallable,
+        TransformerLightningModule,
+    )
+    from rectools.models.nn.transformers.bert4rec import MASKING_VALUE, BERT4RecDataPreparator, ValMaskCallable
+except ImportError:
+    pass
 from tests.models.data import DATASET
 from tests.models.utils import (
     assert_default_config_and_default_model_params_are_the_same,
@@ -41,6 +50,8 @@
 
 from .utils import custom_trainer, leave_one_out_mask
 
+pytestmark = pytest.mark.skipif(sys.version_info >= (3, 13), reason="`torch` is not compatible with Python >= 3.13")
+
 
 class TestBERT4RecModel:
     def setup_method(self) -> None:

tests/models/nn/transformers/test_data_preparator.py

@@ -12,6 +12,7 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
+import sys
 import typing as tp
 
 import numpy as np
@@ -21,9 +22,15 @@
 from rectools.columns import Columns
 from rectools.dataset import Dataset, IdMap, Interactions
 from rectools.dataset.features import DenseFeatures
-from rectools.models.nn.transformers.data_preparator import SequenceDataset, TransformerDataPreparatorBase
+
+try:
+    from rectools.models.nn.transformers.data_preparator import SequenceDataset, TransformerDataPreparatorBase
+except ImportError:
+    pass
 from tests.testing_utils import assert_feature_set_equal, assert_id_map_equal, assert_interactions_set_equal
 
+pytestmark = pytest.mark.skipif(sys.version_info >= (3, 13), reason="`torch` is not compatible with Python >= 3.13")
+
 
 class TestSequenceDataset:
 

tests/models/nn/transformers/test_sasrec.py

@@ -14,27 +14,36 @@
 
 # pylint: disable=too-many-lines
 
+import sys
 import typing as tp
 from functools import partial
 
 import numpy as np
 import pandas as pd
 import pytest
-import torch
-from pytorch_lightning import Trainer, seed_everything
+
+try:
+    import torch
+    from pytorch_lightning import Trainer, seed_everything
+except ImportError:
+    pass
 
 from rectools import ExternalIds
 from rectools.columns import Columns
 from rectools.dataset import Dataset, IdMap, Interactions
-from rectools.models import SASRecModel
-from rectools.models.nn.item_net import CatFeaturesItemNet, IdEmbeddingsItemNet, SumOfEmbeddingsConstructor
-from rectools.models.nn.transformers.base import (
-    LearnableInversePositionalEncoding,
-    TrainerCallable,
-    TransformerLightningModule,
-    TransformerTorchBackbone,
-)
-from rectools.models.nn.transformers.sasrec import SASRecDataPreparator, SASRecTransformerLayers
+
+try:
+    from rectools.models import SASRecModel
+    from rectools.models.nn.item_net import CatFeaturesItemNet, IdEmbeddingsItemNet, SumOfEmbeddingsConstructor
+    from rectools.models.nn.transformers.base import (
+        LearnableInversePositionalEncoding,
+        TrainerCallable,
+        TransformerLightningModule,
+        TransformerTorchBackbone,
+    )
+    from rectools.models.nn.transformers.sasrec import SASRecDataPreparator, SASRecTransformerLayers
+except ImportError:
+    pass
 from tests.models.data import DATASET
 from tests.models.utils import (
     assert_default_config_and_default_model_params_are_the_same,
@@ -44,6 +53,8 @@
 
 from .utils import custom_trainer, leave_one_out_mask
 
+pytestmark = pytest.mark.skipif(sys.version_info >= (3, 13), reason="`torch` is not compatible with Python >= 3.13")
+
 
 class TestSASRecModel:
     def setup_method(self) -> None:

tests/models/nn/transformers/utils.py

@@ -12,12 +12,21 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
+import sys
+
 import pandas as pd
-from pytorch_lightning import Trainer
-from pytorch_lightning.callbacks import ModelCheckpoint
+import pytest
+
+try:
+    from pytorch_lightning import Trainer
+    from pytorch_lightning.callbacks import ModelCheckpoint
+except ImportError:
+    pass
 
 from rectools import Columns
 
+pytestmark = pytest.mark.skipif(sys.version_info >= (3, 13), reason="`torch` is not compatible with Python >= 3.13")
+
 
 def leave_one_out_mask(interactions: pd.DataFrame) -> pd.Series:
     rank = (

tests/models/rank/test_rank.py

@@ -12,15 +12,26 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
+import sys
 import typing as tp
 from itertools import product
 
 import numpy as np
 import pytest
-import torch
+
+try:
+    import torch
+except ImportError:
+    pass
 from scipy import sparse
 
-from rectools.models.rank import Distance, ImplicitRanker, Ranker, TorchRanker
+from rectools.models.rank import Distance, ImplicitRanker, Ranker
+
+try:
+    from rectools.models.rank import TorchRanker
+except ImportError:
+    TorchRanker = object  # it's ok in case we're skipping the tests
+
 
 T = tp.TypeVar("T")
 EPS_DIGITS = 5
@@ -45,7 +56,10 @@ def gen_rankers() -> tp.List[tp.Tuple[tp.Any, tp.Dict[str, tp.Any]]]:
     )
     implicit_ranker_args = [(ImplicitRanker, dict(zip(implicit_keys, v))) for v in implicit_vals]
 
-    return [*torch_ranker_args, *implicit_ranker_args]
+    if sys.version_info >= (3, 13):
+        return implicit_ranker_args
+    else:
+        return [*torch_ranker_args, *implicit_ranker_args]
 
 
 class TestRanker:  # pylint: disable=protected-access

tests/models/rank/test_rank_torch.py

@@ -12,19 +12,28 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
+import sys
 import typing as tp
 from itertools import product
 
 import numpy as np
 import pytest
-import torch
+
+try:
+    import torch
+except ImportError:
+    pass
 from scipy import sparse
 
-from rectools.models.rank import Distance, Ranker, TorchRanker
+try:
+    from rectools.models.rank import Distance, Ranker, TorchRanker
+except ImportError:
+    pass
 
 T = tp.TypeVar("T")
 EPS_DIGITS = 5
 pytestmark = pytest.mark.filterwarnings("ignore:invalid value encountered in true_divide")
+pytestmark = pytest.mark.skipif(sys.version_info >= (3, 13), reason="`torch` is not compatible with Python >= 3.13")
 
 
 def gen_rankers() -> tp.List[tp.Tuple[tp.Any, tp.Dict[str, tp.Any]]]:

tests/models/test_serialization.py

@@ -29,8 +29,12 @@
     LightFM = object  # it's ok in case we're skipping the tests
 
 from rectools.metrics import NDCG
+
+try:
+    from rectools.models import DSSMModel
+except ImportError:
+    DMMSModel = object  # it's ok in case we're skipping the tests
 from rectools.models import (
-    DSSMModel,
     EASEModel,
     ImplicitALSWrapperModel,
     ImplicitBPRWrapperModel,
@@ -44,7 +48,11 @@
     serialization,
 )
 from rectools.models.base import ModelBase, ModelConfig
-from rectools.models.nn.transformers.base import TransformerModelBase
+
+try:
+    from rectools.models.nn.transformers.base import TransformerModelBase
+except ImportError:
+    TransformerModelBase = object
 from rectools.models.vector import VectorModel
 from rectools.utils.config import BaseConfig