Restore pydash in ux test flow to verify conda env package installation

pydash is intentionally used to prove the conda environment properly
resolves and installs packages, not just that code packaging works.
Add pydash to both @conda steps (start and test_functions) since
test_functions calls the functions that import pydash at runtime.

Author: Nissan Pow

tests/functions/ux/flows/function_module.py

@@ -8,9 +8,12 @@
 def avro_transform_string(
     data: str, params: Optional[FunctionParameters] = None
 ) -> str:
-    """Simple avro function that transforms a string"""
+    """Simple avro function that transforms a string using pydash"""
+    import pydash as _
+
     suffix = params.suffix if params and hasattr(params, "suffix") else "default"
-    return data.upper() + "_" + str(suffix)
+    # Use pydash to uppercase the string
+    return _.upper_case(data).replace(" ", "") + "_" + str(suffix)
 
 
 @avro_function
@@ -33,6 +36,10 @@ def json_transform_list(
 def json_process_object(
     data: dict, params: Optional[FunctionParameters] = None
 ) -> dict:
-    """JSON function that adds fields to a dict"""
+    """JSON function that adds a field using pydash"""
+    import pydash as _
+
+    # Use pydash to merge the data with new fields
     increment = params.increment if params and hasattr(params, "increment") else 1
-    return {**data, "processed": True, "increment": increment}
+    result = _.merge({}, data, {"processed": True, "increment": increment})
+    return result

tests/functions/ux/flows/hellosimplefunction.py

@@ -2,7 +2,7 @@
 
 
 class HelloSimpleFunction(FlowSpec):
-    @conda
+    @conda(libraries={"pydash": "5.1.0"})
     @step
     def start(self):
         # Set parameters - these will be used by the functions
@@ -30,7 +30,7 @@ def bind_functions(self):
 
         self.next(self.test_functions)
 
-    @conda
+    @conda(libraries={"pydash": "5.1.0"})
     @step
     def test_functions(self):
         from metaflow_extensions.nflx.plugins.functions.core.function import (