[Docs] Update usage of 'to_mock_server' (#5503)

mlrun · May 5, 2024 · 850d4f7 · 850d4f7
1 parent 3867fa5
commit 850d4f7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/serving/writing-custom-steps.ipynb b/docs/serving/writing-custom-steps.ipynb
@@ -159,6 +159,7 @@
     "echo_server = fn_echo.to_mock_server(current_function=\"*\")\n",
     "\n",
     "result = echo_server.test(\"\", {\"inputs\": 123})\n",
+    "echo_server.wait_for_completion()\n",
     "\n",
     "print(result)"
    ]