[MRG] Solve example throwing an error when executed on a GPU by ncassereau · Pull Request #391 · PythonOT/POT

Types of changes

Changing code for backend example

Motivation and context / Related issue

Resolves #389

How has this been tested (if it applies)

Tested on Jean Zay with a V100

PR checklist

  • I have read the CONTRIBUTING document.
  • The documentation is up-to-date with the changes I made (check build artifacts).
  • All tests passed, and additional code has been covered with new tests.
  • I have added the PR and Issue fix to the RELEASES.md file.