Skip to content

Commit d63e84e

Browse files
committed
Update _transformers.py
1 parent c6210ef commit d63e84e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

autointent/configs/_transformers.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -123,19 +123,19 @@ class CrossEncoderConfig(HFModelConfig):
123123
default_factory=lambda: TokenizerConfig(max_length=512)
124124
) # this is because sentence-transformers doesn't allow you to customize tokenizer settings properly
125125

126-
class СNNConfig(BaseModel):
126+
class CNNConfig(BaseModel):
127127
model_config = ConfigDict(extra="forbid")
128128
max_seq_length: int = Field(128, description="Maximum sequence length.")
129129
padding_idx: int = Field(0, description="Index used for padding.")
130130
unknown_idx: int = Field(1, description="Index used for unknown.")
131131
batch_size: PositiveInt = Field(32, description="Batch size for model inference.")
132132

133133
@classmethod
134-
def from_search_config(cls, values: dict[str, Any] | str | BaseModel | None) -> "СNNConfig":
134+
def from_search_config(cls, values: dict[str, Any] | str | BaseModel | None) -> "CNNConfig":
135135
if values is None:
136136
return cls()
137137
if isinstance(values, BaseModel):
138138
return values # type: ignore[return-value]
139139
if isinstance(values, str):
140140
return cls()
141-
return cls(**values)
141+
return cls(**values)

0 commit comments

Comments
 (0)