Add numpy backend for examples with visualizations

geomstats · Dec 31, 2018 · f917dad · f917dad
1 parent 4f23249
commit f917dad
Show file tree

Hide file tree

Showing 8 changed files with 24 additions and 0 deletions.
diff --git a/examples/plot_geodesics_h2.py b/examples/plot_geodesics_h2.py
@@ -5,6 +5,9 @@
 
 import matplotlib.pyplot as plt
 import numpy as np
+import os
+# Examples with visualizations are only implemented with numpy
+os.environ['GEOMSTATS_BACKEND'] = 'numpy'  # NOQA
 
 import geomstats.visualization as visualization
 

diff --git a/examples/plot_geodesics_s2.py b/examples/plot_geodesics_s2.py
@@ -4,6 +4,9 @@
 
 import matplotlib.pyplot as plt
 import numpy as np
+import os
+# Examples with visualizations are only implemented with numpy
+os.environ['GEOMSTATS_BACKEND'] = 'numpy'  # NOQA
 
 import geomstats.visualization as visualization
 

diff --git a/examples/plot_geodesics_se3.py b/examples/plot_geodesics_se3.py
@@ -5,6 +5,9 @@
 
 import matplotlib.pyplot as plt
 import numpy as np
+import os
+# Examples with visualizations are only implemented with numpy
+os.environ['GEOMSTATS_BACKEND'] = 'numpy'  # NOQA
 
 import geomstats.visualization as visualization
 

diff --git a/examples/plot_geodesics_so3.py b/examples/plot_geodesics_so3.py
@@ -5,6 +5,9 @@
 
 import matplotlib.pyplot as plt
 import numpy as np
+import os
+# Examples with visualizations are only implemented with numpy
+os.environ['GEOMSTATS_BACKEND'] = 'numpy'  # NOQA
 
 import geomstats.visualization as visualization
 

diff --git a/examples/plot_grid_h2.py b/examples/plot_grid_h2.py
@@ -5,6 +5,9 @@
 
 import matplotlib.pyplot as plt
 import numpy as np
+import os
+# Examples with visualizations are only implemented with numpy
+os.environ['GEOMSTATS_BACKEND'] = 'numpy'  # NOQA
 
 import geomstats.visualization as visualization
 

diff --git a/examples/plot_square_h2_klein_disk.py b/examples/plot_square_h2_klein_disk.py
@@ -4,6 +4,9 @@
 
 import matplotlib.pyplot as plt
 import numpy as np
+import os
+# Examples with visualizations are only implemented with numpy
+os.environ['GEOMSTATS_BACKEND'] = 'numpy'  # NOQA
 
 import geomstats.visualization as visualization
 

diff --git a/examples/plot_square_h2_poincare_disk.py b/examples/plot_square_h2_poincare_disk.py
@@ -4,6 +4,9 @@
 
 import matplotlib.pyplot as plt
 import numpy as np
+import os
+# Examples with visualizations are only implemented with numpy
+os.environ['GEOMSTATS_BACKEND'] = 'numpy'  # NOQA
 
 import geomstats.visualization as visualization
 

diff --git a/examples/plot_square_h2_poincare_half_plane.py b/examples/plot_square_h2_poincare_half_plane.py
@@ -4,6 +4,9 @@
 
 import matplotlib.pyplot as plt
 import numpy as np
+import os
+# Examples with visualizations are only implemented with numpy
+os.environ['GEOMSTATS_BACKEND'] = 'numpy'  # NOQA
 
 import geomstats.visualization as visualization