Move import of egoio to if statement to avoid import on rtd

etrago/tools/constraints.py

@@ -21,9 +21,9 @@
 """
 Constraints.py includes additional constraints for eTraGo-optimizations
 """
+import os
 import logging
 
-from egoio.tools import db
 from pyomo.environ import Constraint
 from pypsa.descriptors import expand_series
 from pypsa.linopt import define_constraints, define_variables, get_var, linexpr
@@ -32,6 +32,9 @@
 import pandas as pd
 import pyomo.environ as po
 
+if "READTHEDOCS" not in os.environ:
+    from egoio.tools import db
+
 logger = logging.getLogger(__name__)
 
 __copyright__ = (

etrago/tools/network.py

@@ -22,13 +22,16 @@
 Define class Etrago
 """
 
+import os
 import logging
 
-from egoio.tools import db
 from pypsa.components import Network
 from sqlalchemy.orm import sessionmaker
 import pandas as pd
 
+if "READTHEDOCS" not in os.environ:
+    from egoio.tools import db
+
 from etrago import __version__
 from etrago.cluster.disaggregation import run_disaggregation
 from etrago.cluster.electrical import ehv_clustering, run_spatial_clustering

etrago/tools/utilities.py

@@ -29,7 +29,6 @@
 import math
 import os
 
-from egoio.tools import db
 from pyomo.environ import Constraint, PositiveReals, Var
 from shapely.geometry import Point
 import geopandas as gpd
@@ -38,6 +37,9 @@
 import pypsa
 import sqlalchemy.exc
 
+if "READTHEDOCS" not in os.environ:
+    from egoio.tools import db
+
 logger = logging.getLogger(__name__)