Merge pull request #2077 from recommenders-team/simonz/issue-2064-nam…

…eerror-implicitcf Catch import error separately for SUMModel
recommenders-team · Apr 4, 2024 · a739e75 · a739e75
2 parents 96fe3cf + bd1a95f
commit a739e75
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/smoke/recommenders/recommender/test_deeprec_model.py b/tests/smoke/recommenders/recommender/test_deeprec_model.py
@@ -20,7 +20,6 @@
     from recommenders.models.deeprec.io.dkn_iterator import DKNTextIterator
     from recommenders.models.deeprec.io.sequential_iterator import SequentialIterator
     from recommenders.models.deeprec.models.sequential.sli_rec import SLI_RECModel
-    from recommenders.models.deeprec.models.sequential.sum import SUMModel
     from recommenders.datasets.amazon_reviews import (
         download_and_extract,
         data_preprocessing,
@@ -31,6 +30,11 @@
 except ImportError:
     pass  # disable error while collecting tests for non-gpu environments
 
+try:
+    from recommenders.models.deeprec.models.sequential.sum import SUMModel
+except ImportError:
+    pass  # disable error while collecting tests for SUMModel
+
 
 @pytest.mark.gpu
 def test_FFM_iterator(deeprec_resource_path):