add conditional to not import GUI code in readthedocs

CraigYanitski · Oct 18, 2023 · f2299e4 · f2299e4
1 parent 344d158
commit f2299e4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/kosmatau3d/models/plotting/__init__.py b/kosmatau3d/models/plotting/__init__.py
@@ -1,12 +1,14 @@
 from pprint import pprint
 
 import numpy as np
+import os
 from astropy.io import fits
 import matplotlib.pyplot as plt
 import matplotlib.colors as colors
 from scipy.interpolate import interp2d
 
-from .viewMap import *
+if "READTHEDOCS" not in os.environ:
+    from .viewMap import *
 from .diagram import *
 from kosmatau3d.models import constants