Skip to content

Commit ee3b554

Browse files
committed
consistent naming for esm2
1 parent 6b097cc commit ee3b554

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

chebai_proteins/preprocessing/datasets/deepGO/go_uniprot.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ def __init__(
119119
use_esm2_embeddings: bool = False,
120120
**kwargs,
121121
):
122-
if use_esm2_embeddings:
122+
if bool(use_esm2_embeddings):
123123
self.READER = dr.ESM2EmbeddingReader
124124

125125
self.go_branch: str = self._get_go_branch(go_branch)

chebai_proteins/preprocessing/datasets/scope/scope.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,10 +74,10 @@ def __init__(
7474
scope_version: str,
7575
scope_version_train: Optional[str] = None,
7676
max_sequence_len: int = 1002,
77-
use_esm2_embedding: bool = False,
77+
use_esm2_embeddings: bool = False,
7878
**kwargs,
7979
):
80-
if use_esm2_embedding:
80+
if bool(use_esm2_embeddings):
8181
self.READER = ESM2EmbeddingReader
8282

8383
self.scope_version: str = scope_version

0 commit comments

Comments
 (0)