Bugfix

zillow · Aug 19, 2024 · 06dac92 · 06dac92
1 parent 5411035
commit 06dac92
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/metaflow/plugins/__init__.py b/metaflow/plugins/__init__.py
@@ -161,7 +161,6 @@ def get_plugin_cli():
 from .aws.step_functions.schedule_decorator import ScheduleDecorator
 from .project_decorator import ProjectDecorator
 from .aip.s3_sensor_decorator import S3SensorDecorator
-
 from .aip.exit_handler_decorator import ExitHandlerDecorator
 
 FLOW_DECORATORS = [

diff --git a/metaflow/plugins/aip/tests/flows/skip_flow.py b/metaflow/plugins/aip/tests/flows/skip_flow.py
@@ -18,13 +18,14 @@ def skipped_step(self):
         raise Exception(
             "Unexpectedly ran the skipped_step step. This step should have been skipped."
         )
-        self.next(self.unreachable)
+        self.next(self.unreachable_step)
 
-    def unreachable(self):
+    @step
+    def unreachable_step(self):
         raise Exception(
             "Unexpectedly ran the unreachable step. This step should have been skipped."
         )
-        self.next(self.end)
+        self.next(self.desired_step)
 
     @skip(check="condition_false", next="end")
     @step