Skip to content

Commit bc7fa10

Browse files
committed
fix: remove type hints from private module
1 parent c05614c commit bc7fa10

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

flair/training_utils.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010

1111
from numpy import ndarray
1212
from scipy.stats import pearsonr, spearmanr
13-
from scipy.stats._stats_py import PearsonRResult, SignificanceResult
1413
from sklearn.metrics import mean_absolute_error, mean_squared_error
1514
from torch.optim import Optimizer
1615
from torch.utils.data import Dataset
@@ -61,10 +60,10 @@ def mean_squared_error(self) -> Union[float, ndarray]:
6160
def mean_absolute_error(self):
6261
return mean_absolute_error(self.true, self.pred)
6362

64-
def pearsonr(self) -> PearsonRResult:
63+
def pearsonr(self):
6564
return pearsonr(self.true, self.pred)[0]
6665

67-
def spearmanr(self) -> SignificanceResult:
66+
def spearmanr(self):
6867
return spearmanr(self.true, self.pred)[0]
6968

7069
def to_tsv(self) -> str:

0 commit comments

Comments
 (0)