diff options
Diffstat (limited to 'test/test_dr.py')
-rw-r--r-- | test/test_dr.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/test/test_dr.py b/test/test_dr.py index 3da7705..3faba48 100644 --- a/test/test_dr.py +++ b/test/test_dr.py @@ -1,8 +1,9 @@ -import ot + import numpy as np +import ot import pytest -try: # test if cudamat installed +try: # test if autograd and pymanopt are installed import ot.dr nogo = False except ImportError: |