We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6b097cc commit ee3b554Copy full SHA for ee3b554
2 files changed
chebai_proteins/preprocessing/datasets/deepGO/go_uniprot.py
@@ -119,7 +119,7 @@ def __init__(
119
use_esm2_embeddings: bool = False,
120
**kwargs,
121
):
122
- if use_esm2_embeddings:
+ if bool(use_esm2_embeddings):
123
self.READER = dr.ESM2EmbeddingReader
124
125
self.go_branch: str = self._get_go_branch(go_branch)
chebai_proteins/preprocessing/datasets/scope/scope.py
@@ -74,10 +74,10 @@ def __init__(
74
scope_version: str,
75
scope_version_train: Optional[str] = None,
76
max_sequence_len: int = 1002,
77
- use_esm2_embedding: bool = False,
+ use_esm2_embeddings: bool = False,
78
79
80
- if use_esm2_embedding:
81
self.READER = ESM2EmbeddingReader
82
83
self.scope_version: str = scope_version
0 commit comments