mirror of
https://github.com/Findus23/collision-analyisis-and-interpolation.git
synced 2024-09-19 15:13:50 +02:00
improve custom scaler typing
This commit is contained in:
parent
0a5f4f261e
commit
88bbf9a160
1 changed files with 2 additions and 2 deletions
|
@ -1,4 +1,4 @@
|
|||
from typing import List
|
||||
from typing import List, Iterator
|
||||
|
||||
import numpy as np
|
||||
|
||||
|
@ -28,7 +28,7 @@ class CustomScaler:
|
|||
return (data - self.means) / self.stds
|
||||
# return data
|
||||
|
||||
def transform_parameters(self, parameters: List) -> List:
|
||||
def transform_parameters(self, parameters: List) -> Iterator[float]:
|
||||
self._check_fitted()
|
||||
if len(parameters) != len(self.means):
|
||||
raise ValueError("incorrect number of parameters")
|
||||
|
|
Loading…
Reference in a new issue