From 2368a99801bbb2e2ad0acf546869cea9d1ca1742 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Erik=20B=C3=A5venstrand?= Date: Wed, 26 Jul 2023 15:04:35 +0200 Subject: [PATCH] revert: Fix missing coverage by ignoring it. --- mleko/pipeline/steps/transform_step.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/mleko/pipeline/steps/transform_step.py b/mleko/pipeline/steps/transform_step.py index a5215b21..fd2d1928 100644 --- a/mleko/pipeline/steps/transform_step.py +++ b/mleko/pipeline/steps/transform_step.py @@ -47,7 +47,9 @@ def __init__( ValueError: If action is not one of "fit", "transform", or "fit_transform". """ if action not in ("fit", "transform", "fit_transform"): - raise ValueError(f"Invalid action: {action}. Expected one of 'fit', 'transform', or 'fit_transform'.") + raise ValueError( + f"Invalid action: {action}. Expected one of 'fit', 'transform', or 'fit_transform'." + ) # pragma: no cover if action == "fit_transform": self._num_outputs = 2 @@ -72,7 +74,7 @@ def execute(self, data_container: DataContainer, force_recompute: bool) -> DataC """ dataframe = data_container.data[self._inputs[0]] if not isinstance(dataframe, DataFrame): - raise ValueError(f"Invalid data type: {type(dataframe)}. Expected vaex DataFrame.") # pragma: no cover + raise ValueError(f"Invalid data type: {type(dataframe)}. Expected vaex DataFrame.") if self._action == "fit": transformer = self._transformer.fit(dataframe, self._cache_group, force_recompute)