Add message for examples in TF

examples/plot_geodesics_h2.py

@@ -6,8 +6,6 @@
 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
 
@@ -64,4 +62,10 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    if os.environ['GEOMSTATS_BACKEND'] == 'tensorflow':
+        print('Examples with visualizations are only implemented '
+              'with numpy backend.\n'
+              'To change backend, write: '
+              'export GEOMSTATS_BACKEND = \'numpy\'.')
+    else:
+        main()

examples/plot_geodesics_s2.py

@@ -5,8 +5,6 @@
 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
 
@@ -33,4 +31,10 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    if os.environ['GEOMSTATS_BACKEND'] == 'tensorflow':
+        print('Examples with visualizations are only implemented '
+              'with numpy backend.\n'
+              'To change backend, write: '
+              'export GEOMSTATS_BACKEND = \'numpy\'.')
+    else:
+        main()

examples/plot_geodesics_se3.py

@@ -6,8 +6,6 @@
 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
 
@@ -33,4 +31,10 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    if os.environ['GEOMSTATS_BACKEND'] == 'tensorflow':
+        print('Examples with visualizations are only implemented '
+              'with numpy backend.\n'
+              'To change backend, write: '
+              'export GEOMSTATS_BACKEND = \'numpy\'.')
+    else:
+        main()

examples/plot_geodesics_so3.py

@@ -6,8 +6,6 @@
 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
 
@@ -32,4 +30,10 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    if os.environ['GEOMSTATS_BACKEND'] == 'tensorflow':
+        print('Examples with visualizations are only implemented '
+              'with numpy backend.\n'
+              'To change backend, write: '
+              'export GEOMSTATS_BACKEND = \'numpy\'.')
+    else:
+        main()

examples/plot_grid_h2.py

@@ -6,8 +6,6 @@
 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
 
@@ -46,4 +44,10 @@ def main(left=-128,
 
 
 if __name__ == "__main__":
-    main()
+    if os.environ['GEOMSTATS_BACKEND'] == 'tensorflow':
+        print('Examples with visualizations are only implemented '
+              'with numpy backend.\n'
+              'To change backend, write: '
+              'export GEOMSTATS_BACKEND = \'numpy\'.')
+    else:
+        main()

examples/plot_quantization_s1.py

@@ -4,6 +4,7 @@
 """
 
 import matplotlib.pyplot as plt
+import os
 
 import geomstats.visualization as visualization
 
@@ -39,4 +40,10 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    if os.environ['GEOMSTATS_BACKEND'] == 'tensorflow':
+        print('Examples with visualizations are only implemented '
+              'with numpy backend.\n'
+              'To change backend, write: '
+              'export GEOMSTATS_BACKEND = \'numpy\'.')
+    else:
+        main()

examples/plot_quantization_s2.py

@@ -4,6 +4,7 @@
 """
 
 import matplotlib.pyplot as plt
+import os
 
 import geomstats.visualization as visualization
 
@@ -40,4 +41,10 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    if os.environ['GEOMSTATS_BACKEND'] == 'tensorflow':
+        print('Examples with visualizations are only implemented '
+              'with numpy backend.\n'
+              'To change backend, write: '
+              'export GEOMSTATS_BACKEND = \'numpy\'.')
+    else:
+        main()

examples/plot_square_h2_klein_disk.py

@@ -5,8 +5,6 @@
 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
 
@@ -45,4 +43,10 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    if os.environ['GEOMSTATS_BACKEND'] == 'tensorflow':
+        print('Examples with visualizations are only implemented '
+              'with numpy backend.\n'
+              'To change backend, write: '
+              'export GEOMSTATS_BACKEND = \'numpy\'.')
+    else:
+        main()

examples/plot_square_h2_poincare_disk.py

@@ -5,8 +5,6 @@
 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
 
@@ -47,4 +45,10 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    if os.environ['GEOMSTATS_BACKEND'] == 'tensorflow':
+        print('Examples with visualizations are only implemented '
+              'with numpy backend.\n'
+              'To change backend, write: '
+              'export GEOMSTATS_BACKEND = \'numpy\'.')
+    else:
+        main()

examples/plot_square_h2_poincare_half_plane.py

@@ -5,8 +5,6 @@
 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
 
@@ -47,4 +45,10 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    if os.environ['GEOMSTATS_BACKEND'] == 'tensorflow':
+        print('Examples with visualizations are only implemented '
+              'with numpy backend.\n'
+              'To change backend, write: '
+              'export GEOMSTATS_BACKEND = \'numpy\'.')
+    else:
+        main()

tests/test_examples.py

@@ -20,9 +20,9 @@
 import examples.plot_square_h2_poincare_disk as plot_square_h2_poincare_disk
 import examples.plot_square_h2_poincare_half_plane as plot_square_h2_poincare_half_plane  # NOQA
 import examples.plot_square_h2_klein_disk as plot_square_h2_klein_disk
-import examples.tangent_pca_so3 as tangent_pca_so3
 import examples.plot_quantization_s1 as plot_quantization_s1
 import examples.plot_quantization_s2 as plot_quantization_s2
+import examples.tangent_pca_so3 as tangent_pca_so3
 import geomstats.tests