fix rdkit version check on install

chemosim-lab · Nov 17, 2022 · 65acfa0 · 65acfa0
1 parent a36b470
commit 65acfa0
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -6,15 +6,16 @@
 import versioneer
 
 GITHUB_ACTIONS = os.environ.get("GITHUB_ACTIONS", False)
+READTHEDOCS = os.environ.get("READTHEDOCS", False)
 
 # manually check RDKit version
 try:
     from rdkit import __version__ as rdkit_version
 except ImportError:
-    if not GITHUB_ACTIONS:
+    if not (GITHUB_ACTIONS or READTHEDOCS):
         raise ImportError("ProLIF requires RDKit but it is not installed")
 else:
-    if re.match(r"^20[0-1][0-9]\.", rdkit_version):
-        raise ValueError("ProLIF requires a version of RDKit >= 2020")
+    if re.match(r"^(20[0-1][0-9])|(2020)", rdkit_version):
+        raise ValueError("ProLIF requires a version of RDKit >= 2021")
 
 setup(version=versioneer.get_version())