remove doctest for beckend tests

Author: rflamary

.github/workflows/build_tests.yml

@@ -155,7 +155,7 @@ jobs:
         pip install pytest pytest-cov
     - name: Run tests
       run: |
-        python -m pytest --durations=20 -v test/ ot/ --doctest-modules --color=yes --cov=./ --cov-report=xml
+        python -m pytest --durations=20 -v test/ --color=yes --cov=./ --cov-report=xml
     - name: Upload coverage reports to Codecov with GitHub Action
       uses: codecov/codecov-action@v4
 
@@ -183,7 +183,7 @@ jobs:
         pip install pytest pytest-cov
     - name: Run tests
       run: |
-        python -m pytest --durations=20 -v test/ ot/ --doctest-modules --color=yes --cov=./ --cov-report=xml
+        python -m pytest --durations=20 -v test/ --color=yes --cov=./ --cov-report=xml
     - name: Upload coverage reports to Codecov with GitHub Action
       uses: codecov/codecov-action@v4
 
@@ -211,7 +211,7 @@ jobs:
         pip install pytest pytest-cov
     - name: Run tests
       run: |
-        python -m pytest --durations=20 -v test/ ot/ --doctest-modules --color=yes --cov=./ --cov-report=xml
+        python -m pytest --durations=20 -v test/ --color=yes --cov=./ --cov-report=xml
     - name: Upload coverage reports to Codecov with GitHub Action
       uses: codecov/codecov-action@v4
 

ot/__init__.py

@@ -93,7 +93,7 @@
 from .utils import dist, unif, tic, toc, toq
 
 
-__version__ = "0.9.7"
+__version__ = "0.9.7.dev0"
 
 __all__ = [
     "emd",

ot/gnn/_layers.py

@@ -7,14 +7,16 @@
 #         Rémi Flamary <remi.flamary@unice.fr>
 #
 # License: MIT License
-
-import torch
-import torch.nn as nn
-from ._utils import (
-    TFGW_template_initialization,
-    FGW_distance_to_templates,
-    wasserstein_distance_to_templates,
-)
+try:
+    import torch
+    import torch.nn as nn
+    from ._utils import (
+        TFGW_template_initialization,
+        FGW_distance_to_templates,
+        wasserstein_distance_to_templates,
+    )
+except ImportError:
+    pass
 
 
 class TFGWPooling(nn.Module):