update migration script

Author: aditya0by0

chebai_proteins/preprocessing/migration/deep_go/migrate_deep_go_1_data.py

@@ -3,10 +3,11 @@
 from typing import List, Literal, Optional, Tuple
 
 import pandas as pd
-from chebai.preprocessing.datasets.deepGO.go_uniprot import DeepGO1MigratedData
 from iterstrat.ml_stratifiers import MultilabelStratifiedShuffleSplit
 from jsonargparse import CLI
 
+from chebai_proteins.preprocessing.datasets.deepGO.go_uniprot import DeepGO1MigratedData
+
 
 class DeepGo1DataMigration:
     """

chebai_proteins/preprocessing/migration/deep_go/migrate_deep_go_2_data.py

@@ -4,10 +4,11 @@
 from typing import List, Literal, Optional
 
 import pandas as pd
-from chebai.preprocessing.datasets.deepGO.go_uniprot import DeepGO2MigratedData
-from chebai.preprocessing.reader import ProteinDataReader
 from jsonargparse import CLI
 
+from chebai_proteins.preprocessing.datasets.deepGO.go_uniprot import DeepGO2MigratedData
+from chebai_proteins.preprocessing.reader import ProteinDataReader
+
 
 class DeepGo2DataMigration:
     """