TESTS: add .coveragerc

batistagroup · Oct 24, 2023 · efb218d · efb218d
1 parent 799351d
commit efb218d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/.coveragerc b/.coveragerc
@@ -0,0 +1,10 @@
+[run]
+source = 
+    ChemSpaceAL/InitializeWorkspace.py
+
+omit =
+    ChemSpaceAL/Model.py
+    ChemSpaceAL/Dataset.py
+    ChemSpaceAL/Docking.py
+    ChemSpaceAL/Generation.py
+    ChemSpaceAL/Training.py
diff --git a/ChemSpaceAL/Configuration.py b/ChemSpaceAL/Configuration.py
@@ -2,7 +2,7 @@
 from torch.nn import Module
 import os
 from typing import Union, Dict, Any, List, Tuple, Optional, Any, Set, Callable, cast
-from ChemSpaceAL.InitializeWorkspace import FOLDER_STRUCTURE as fldr_struc
+from .InitializeWorkspace import FOLDER_STRUCTURE as fldr_struc
 import textwrap
 from rdkit.Chem import Descriptors
 from rdkit import Chem