fix imports from chebai_proteins

Author: aditya0by0

chebai_proteins/preprocessing/datasets/deepGO/go_uniprot.py

@@ -42,9 +42,10 @@
 import torch
 import tqdm
 from Bio import SwissProt
-from chebai.preprocessing import reader as dr
 from chebai.preprocessing.datasets.base import _DynamicDataset
 
+from chebai_proteins.preprocessing import reader as dr
+
 # fmt: off
 # https://github.com/bio-ontology-research-group/deepgo/blob/master/utils.py#L15
 EXPERIMENTAL_EVIDENCE_CODES = {

chebai_proteins/preprocessing/datasets/deepGO/protein_pretraining.py

@@ -10,13 +10,14 @@
 import torch
 from Bio import SwissProt
 from chebai.preprocessing.datasets.base import _DynamicDataset
-from chebai.preprocessing.datasets.deepGO.go_uniprot import (
+from sklearn.model_selection import train_test_split
+
+from chebai_proteins.preprocessing.datasets.deepGO.go_uniprot import (
     AMBIGUOUS_AMINO_ACIDS,
     EXPERIMENTAL_EVIDENCE_CODES,
     GOUniProtOver250,
 )
-from chebai.preprocessing.reader import ProteinDataReader
-from sklearn.model_selection import train_test_split
+from chebai_proteins.preprocessing.reader import ProteinDataReader
 
 
 class _ProteinPretrainingData(_DynamicDataset, ABC):

tests/unit/readers/testProteinDataReader.py

@@ -12,7 +12,7 @@ class TestProteinDataReader(unittest.TestCase):
 
     @classmethod
     @patch(
-        "chebai.preprocessing.reader.open",
+        "chebai_proteins.preprocessing.reader.open",
         new_callable=mock_open,
         read_data="M\nK\nT\nF\nR\nN",
     )