We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ad24fa7 commit 357752fCopy full SHA for 357752f
1 file changed
chebai/preprocessing/reader.py
@@ -509,7 +509,7 @@ class ESM2EmbeddingReader(DataReader):
509
510
def __init__(
511
self,
512
- save_model_dir: str,
+ save_model_dir: str = os.path.join("data", "esm2_reader"),
513
model_name: str = "esm2_t36_3B_UR50D",
514
device: Optional[torch.device] = None,
515
truncation_length: int = 1022,
@@ -617,7 +617,7 @@ def load_hub_workaround(self, url) -> torch.Tensor:
617
return data
618
619
@staticmethod
620
- def name() -> None:
+ def name() -> str:
621
"""
622
Returns the name of the data reader. This method identifies the specific type of data reader.
623
0 commit comments