Merge pull request #2 from TattaBio/andre

rename dgeb imports
TattaBio · Jul 2, 2024 · 1894ba9 · 1894ba9
2 parents 8dc15d3 + 5f1f8b8
commit 1894ba9
Show file tree

Hide file tree

Showing 6 changed files with 25 additions and 25 deletions.
diff --git a/dgeb/tasks/bigene_mining_tasks.py b/dgeb/tasks/bigene_mining_tasks.py
@@ -6,10 +6,10 @@
 import logging
 from collections import defaultdict
 
-from geb.evaluators import BiGeneMiningEvaluator
-from geb.modality import Modality
-from geb.models import BioSeqTransformer
-from geb.tasks.tasks import Dataset, Task, TaskMetadata, TaskResult
+from dgeb.evaluators import BiGeneMiningEvaluator
+from dgeb.modality import Modality
+from dgeb.models import BioSeqTransformer
+from dgeb.tasks.tasks import Dataset, Task, TaskMetadata, TaskResult
 
 logger = logging.getLogger(__name__)
 

diff --git a/dgeb/tasks/classification_tasks.py b/dgeb/tasks/classification_tasks.py
@@ -9,14 +9,14 @@
 import datasets
 import numpy as np
 
-from geb.eval_utils import merge_split_elem_embeds
-from geb.evaluators import (
+from dgeb.eval_utils import merge_split_elem_embeds
+from dgeb.evaluators import (
     MultiClassMultiOutputKNNClassificationEvaluator,
     logRegClassificationEvaluator,
 )
-from geb.modality import Modality
-from geb.models import BioSeqTransformer
-from geb.tasks.tasks import Dataset, Task, TaskMetadata, TaskResult
+from dgeb.modality import Modality
+from dgeb.models import BioSeqTransformer
+from dgeb.tasks.tasks import Dataset, Task, TaskMetadata, TaskResult
 
 logger = logging.getLogger(__name__)
 

diff --git a/dgeb/tasks/clustering_tasks.py b/dgeb/tasks/clustering_tasks.py
@@ -5,10 +5,10 @@
 import logging
 from collections import defaultdict
 
-from geb.evaluators import ClusteringEvaluator
-from geb.modality import Modality
-from geb.models import BioSeqTransformer
-from geb.tasks.tasks import Dataset, Task, TaskMetadata, TaskResult
+from dgeb.evaluators import ClusteringEvaluator
+from dgeb.modality import Modality
+from dgeb.models import BioSeqTransformer
+from dgeb.tasks.tasks import Dataset, Task, TaskMetadata, TaskResult
 
 logger = logging.getLogger(__name__)
 

diff --git a/dgeb/tasks/eds_tasks.py b/dgeb/tasks/eds_tasks.py
@@ -9,10 +9,10 @@
 import numpy as np
 import pandas as pd
 
-from geb.evaluators import EDSEvaluator
-from geb.modality import Modality
-from geb.models import BioSeqTransformer
-from geb.tasks.tasks import Dataset, Task, TaskMetadata, TaskResult
+from dgeb.evaluators import EDSEvaluator
+from dgeb.modality import Modality
+from dgeb.models import BioSeqTransformer
+from dgeb.tasks.tasks import Dataset, Task, TaskMetadata, TaskResult
 
 logger = logging.getLogger(__name__)
 

diff --git a/dgeb/tasks/pair_classification_tasks.py b/dgeb/tasks/pair_classification_tasks.py
@@ -6,10 +6,10 @@
 import logging
 from collections import defaultdict
 
-from geb.evaluators import PairClassificationEvaluator
-from geb.modality import Modality
-from geb.models import BioSeqTransformer
-from geb.tasks.tasks import Dataset, Task, TaskMetadata, TaskResult
+from dgeb.evaluators import PairClassificationEvaluator
+from dgeb.modality import Modality
+from dgeb.models import BioSeqTransformer
+from dgeb.tasks.tasks import Dataset, Task, TaskMetadata, TaskResult
 
 from ..eval_utils import paired_dataset
 

diff --git a/dgeb/tasks/retrieval_tasks.py b/dgeb/tasks/retrieval_tasks.py
@@ -6,10 +6,10 @@
 import logging
 from collections import defaultdict
 
-from geb.evaluators import RetrievalEvaluator
-from geb.modality import Modality
-from geb.models import BioSeqTransformer
-from geb.tasks.tasks import Dataset, Task, TaskMetadata, TaskResult
+from dgeb.evaluators import RetrievalEvaluator
+from dgeb.modality import Modality
+from dgeb.models import BioSeqTransformer
+from dgeb.tasks.tasks import Dataset, Task, TaskMetadata, TaskResult
 
 logger = logging.getLogger(__name__)