try optional waterbridges import

jamesmkrieger · Aug 10, 2023 · aa943a4 · aa943a4
1 parent 10dfba7
commit aa943a4
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/prody/proteins/__init__.py b/prody/proteins/__init__.py
@@ -216,9 +216,15 @@
 from .starfile import *
 __all__.extend(starfile.__all__)
 
-from . import waterbridges
-from .waterbridges import *
-__all__.extend(waterbridges.__all__)
+try:
+    from . import waterbridges
+    from .waterbridges import *
+except SyntaxError:
+    import logging
+    logger = logging.getLogger()
+    logger.warn("Cannot import waterbridges in python 2")
+else:
+    __all__.extend(waterbridges.__all__)
 
 from . import fixer
 from .fixer import *