Commit 54281fda authored by Amir MOHAMMADI's avatar Amir MOHAMMADI

[transfomer][wrapper] rename the wrapped instance to estimator since mario...

[transfomer][wrapper] rename the wrapped instance to estimator since mario expects that for better dask graphs
parent 7642808d
Pipeline #44404 failed with stage
in 9 minutes and 6 seconds
...@@ -51,11 +51,11 @@ class Wrapper(TransformerMixin, BaseEstimator): ...@@ -51,11 +51,11 @@ class Wrapper(TransformerMixin, BaseEstimator):
def __init__( def __init__(
self, self,
transformer, estimator,
**kwargs, **kwargs,
): ):
super().__init__(**kwargs) super().__init__(**kwargs)
self.transformer = transformer self.estimator = estimator
def transform(self, videos, **kwargs): def transform(self, videos, **kwargs):
transformed_videos = [] transformed_videos = []
...@@ -76,12 +76,12 @@ class Wrapper(TransformerMixin, BaseEstimator): ...@@ -76,12 +76,12 @@ class Wrapper(TransformerMixin, BaseEstimator):
for index in video.indices for index in video.indices
] ]
data = self.transformer.transform(video, **kw) data = self.estimator.transform(video, **kw)
dl, vl = len(data), len(video) dl, vl = len(data), len(video)
if dl != vl: if dl != vl:
raise NotImplementedError( raise NotImplementedError(
f"Length of transformed data ({dl}) using {self.transformer}" f"Length of transformed data ({dl}) using {self.estimator}"
f" is different from the length of input video: {vl}" f" is different from the length of input video: {vl}"
) )
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment