1
0
Fork 0
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:
Lukas Winkler 2021-03-08 15:39:47 +01:00
parent 0a5f4f261e
commit 88bbf9a160
Signed by: lukas
GPG key ID: 54DE4D798D244853

View file

@ -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")