refactor: Update embeddings/pipeline/flair_preprocessing_pipeline.py

ktagowski · asawczyn · web-flow · commit 9e697429cae0 · 2022-04-20T12:47:39.000+02:00
Co-authored-by: Albert &lt;34009816+asawczyn@users.noreply.github.com&gt;
diff --git a/embeddings/pipeline/flair_preprocessing_pipeline.py b/embeddings/pipeline/flair_preprocessing_pipeline.py
@@ -114,8 +114,7 @@ def _get_transformations(
                 ClassEncodeColumnTransformation(column=self.target_column_name)
             )
 
-        to_flair_dataset_transformation = self._get_dataset_transformation(data_loader)
-        if to_flair_dataset_transformation:
+        if to_flair_dataset_transformation := self._get_dataset_transformation(data_loader):
             transformation = transformation.then(to_flair_dataset_transformation)
 
         if self.sample_missing_splits:

Original file line number	Diff line number	Diff line change
`@@ -114,8 +114,7 @@ def _get_transformations(`
`114`	`114`	`ClassEncodeColumnTransformation(column=self.target_column_name)`
`115`	`115`	`)`
`116`	`116`
`117`		`- to_flair_dataset_transformation = self._get_dataset_transformation(data_loader)`
`118`		`- if to_flair_dataset_transformation:`
	`117`	`+ if to_flair_dataset_transformation := self._get_dataset_transformation(data_loader):`
`119`	`118`	`transformation = transformation.then(to_flair_dataset_transformation)`
`120`	`119`
`121`	`120`	`if self.sample_missing_splits:`