revert: Fix missing coverage by ignoring it.

klarna-incubator · Jul 26, 2023 · 2368a99 · 2368a99
1 parent 865e52e
commit 2368a99
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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)