Skip to content

Commit 2389687

Browse files
committed
fix ruff
1 parent 4680c08 commit 2389687

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

autointent/_dump_tools.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,9 @@
88
import joblib
99
import numpy as np
1010
import numpy.typing as npt
11+
import torch
1112
from pydantic import BaseModel
1213
from sklearn.base import BaseEstimator
13-
import torch
1414
from torch import nn
1515

1616
from autointent import Embedder, Ranker, VectorIndex

autointent/modules/scoring/_cnn/textcnn.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,10 +63,10 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
6363
concatenated: torch.Tensor = torch.cat(conved, dim=1)
6464
dropped: torch.Tensor = self.dropout(concatenated)
6565
return self.fc(dropped) # type: ignore[no-any-return]
66-
66+
6767
def load(self, model_path: str) -> None:
6868
"""Load model from saved state.
69-
69+
7070
Args:
7171
model_path: Path to the saved model state dictionary.
7272
"""

0 commit comments

Comments
 (0)