Use PyPi dependency for durable functions integration tests

tests/integration/durable_integ_base.py

@@ -37,15 +37,6 @@ def parameterized_test_name(func, num, params):
     @classmethod
     def build_durable_functions(cls):
         """Run sam build for durable functions."""
-        # Set environment variable for SDK .whl file location
-        whl_path = Path(
-            cls.test_data_path,
-            "durable",
-            "functions",
-            "aws_durable_execution_sdk_python-1.0.0-py3-none-any.whl",
-        )
-        os.environ["DURABLE_SDK_WHL"] = str(whl_path.absolute())
-
         cls.build_dir = Path(cls.test_data_path, "durable", ".aws-sam", "build")
         cls.built_template_path = cls.build_dir / "template.yaml"
 

tests/integration/testdata/durable/functions/requirements.txt

@@ -1 +1 @@
-${DURABLE_SDK_WHL}
+aws-durable-execution-sdk-python

tests/integration/testdata/durable/functions/wait_for_callback/wait_for_callback.py

@@ -2,15 +2,18 @@
 from typing import Any
 
 from aws_durable_execution_sdk_python.config import WaitForCallbackConfig
-from aws_durable_execution_sdk_python.context import DurableContext
+from aws_durable_execution_sdk_python.context import (
+    DurableContext,
+    WaitForCallbackContext,
+)
 from aws_durable_execution_sdk_python.execution import durable_execution
 from aws_durable_execution_sdk_python.config import Duration
 
 logger = logging.getLogger()
 logger.setLevel(logging.INFO)
 
 
-def external_system_call(callback_id: str) -> None:
+def external_system_call(callback_id: str, _context: WaitForCallbackContext) -> None:
     """Simulate calling an external system with callback ID."""
     # In real usage, this would make an API call to an external system
     # passing the callback_id for the system to call back when done