fix test

microsoft · Nov 2, 2023 · c83d4eb · c83d4eb
1 parent c6a16a3
commit c83d4eb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/promptflow/promptflow/_sdk/operations/_test_submitter.py b/src/promptflow/promptflow/_sdk/operations/_test_submitter.py
@@ -32,6 +32,7 @@ def __init__(self, flow: Flow, flow_context: FlowContext, client=None):
         self.flow = flow
         self._origin_flow = flow
         self._dataplane_flow = None
+        self.flow_context = flow_context
         # TODO: remove this
         self._variant = flow_context.variant
         from .._pf_client import PFClient