NVIDIA-Merlin / models

Merlin Models is a collection of deep learning recommender system model reference implementations
https://nvidia-merlin.github.io/models/main/index.html
Apache License 2.0
262 stars 50 forks source link

Add importorskip for pytorch_lightning to torch tests #1153

Closed oliverholworthy closed 1 year ago

oliverholworthy commented 1 year ago

Fixes Error in gpu-multi (example where pytest fails to collect tests if pytorch_lighting is not available.

Add importorskip for pytorch_lightning to torch tests module enabling tests to collect in an environment with torch installed but not pytorch_lighting.

github-actions[bot] commented 1 year ago

Documentation preview

https://nvidia-merlin.github.io/models/review/pr-1153