done

fusedrug/data/interaction/drug_target/loaders/dti_binding_dataset_loader.py

@@ -1,6 +1,7 @@
 from typing import List, Optional, Union
 from fuse.utils import NDict
-from fuse.data import OpBase, get_sample_id
+from fuse.data import OpBase
+from fuse.data.utils.sample import get_sample_id
 import fusedrug.data.interaction.drug_target.datasets.dti_binding_dataset as dbd
 
 

fusedrug/data/interaction/drug_target/loaders/indexed_text_PPI_table_loader.py

@@ -1,6 +1,7 @@
 from typing import Optional, Dict
 from fuse.utils import NDict
-from fuse.data import OpBase, get_sample_id
+from fuse.data import OpBase
+from fuse.data.utils.sample import get_sample_id
 from fusedrug.utils.file_formats import IndexedTextTable
 import numpy
 

fusedrug/data/molecule/ops/augment.py

@@ -1,7 +1,8 @@
 import numpy as np
 from rdkit import Chem
 from fuse.utils import NDict
-from fuse.data import OpBase, get_sample_id
+from fuse.data import OpBase
+from fuse.data.utils.sample import get_sample_id
 
 
 class SmilesRandomizeAtomOrder(OpBase):

fusedrug/data/molecule/ops/loaders/smi_loader.py

@@ -1,6 +1,7 @@
 from typing import Optional
 from fuse.utils import NDict
-from fuse.data import OpBase, get_sample_id
+from fuse.data import OpBase
+from fuse.data.utils.sample import get_sample_id
 from fusedrug.utils.file_formats import IndexedTextTable
 
 

fusedrug/data/molecule/ops/smiles_ops.py

@@ -2,7 +2,8 @@
 from rdkit import Chem
 from rdkit.Chem.rdmolops import SanitizeFlags
 from fuse.utils import NDict
-from fuse.data import OpBase, get_sample_id
+from fuse.data import OpBase
+from fuse.data.utils.sample import get_sample_id
 
 
 class SmilesToRDKitMol(OpBase):

fusedrug/data/protein/ops/loaders/fasta_loader.py

@@ -1,6 +1,7 @@
 from typing import Dict, Optional
 from fuse.utils import NDict
-from fuse.data import OpBase, get_sample_id
+from fuse.data import OpBase
+from fuse.data.utils.sample import get_sample_id
 from fusedrug.utils.file_formats import IndexedFasta, IndexedFastaCustom
 
 """

fusedrug/data/tokenizer/ops/fast_tokenizer_ops.py

@@ -1,5 +1,6 @@
 from fuse.utils import NDict
-from fuse.data import OpBase, get_sample_id
+from fuse.data import OpBase
+from fuse.data.utils.sample import get_sample_id
 from tokenizers import Tokenizer
 from tokenizers.normalizers import Normalizer
 from tokenizers.pre_tokenizers import PreTokenizer