Refactor pipeline functions to return specific values

Author: htahir1

pipelines/hello_pipeline/hello_pipeline.py

@@ -16,8 +16,9 @@ def say_hello() -> str:
 
 
 @pipeline
-def hello_pipeline():
-    say_hello()
+def hello_pipeline() -> str:
+    message = say_hello()
+    return message
 
 
 if __name__ == "__main__":

pipelines/io_pipeline/io_pipeline.py

@@ -31,9 +31,10 @@ def count_rows(
 
 
 @pipeline
-def io_pipeline():
+def io_pipeline() -> int:
     features, labels = load_data()
-    count_rows(features, labels)
+    row_count = count_rows(features, labels)
+    return row_count
 
 
 if __name__ == "__main__":

pipelines/meta_pipeline/meta_pipeline.py

@@ -21,8 +21,9 @@ def compute_accuracy() -> Annotated[float, "accuracy_metric"]:
 
 
 @pipeline
-def meta_pipeline():
-    compute_accuracy()
+def meta_pipeline() -> float:
+    accuracy = compute_accuracy()
+    return accuracy
 
 
 if __name__ == "__main__":

pipelines/param_pipeline/param_pipeline.py

@@ -19,8 +19,9 @@ def multiply(number: int, factor: int = 2) -> Annotated[int, "product"]:
 
 
 @pipeline
-def param_pipeline(number: int = 3, factor: int = 2):
-    multiply(number=number, factor=factor)
+def param_pipeline(number: int = 3, factor: int = 2) -> int:
+    result = multiply(number=number, factor=factor)
+    return result
 
 
 if __name__ == "__main__":