Skip to content

Commit

Permalink
Merge pull request #2077 from recommenders-team/simonz/issue-2064-nam…
Browse files Browse the repository at this point in the history
…eerror-implicitcf

Catch import error separately for SUMModel
  • Loading branch information
miguelgfierro authored Apr 4, 2024
2 parents 96fe3cf + bd1a95f commit a739e75
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion tests/smoke/recommenders/recommender/test_deeprec_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand All @@ -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):
Expand Down

0 comments on commit a739e75

Please sign in to comment.