File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
chebai_proteins/preprocessing/datasets Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff 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 )
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments