.

anyscale · Sep 20, 2023 · 61f06fc · 61f06fc
1 parent e5d5ed7
commit 61f06fc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/docs/courses/mlops/jobs-and-services.md b/docs/courses/mlops/jobs-and-services.md
@@ -157,6 +157,11 @@ Similar to Anyscale Jobs, we'll start by creating a [`serve_model.py`](https://g
 ```python linenums="1"
 # deploy/services/serve_model.py
 
+import os
+import subprocess
+from madewithml.config import MODEL_REGISTRY  # NOQA: E402
+from madewithml.serve import ModelDeployment  # NOQA: E402
+
 # Copy from S3
 github_username = os.environ.get("GITHUB_USERNAME")
 subprocess.check_output(["aws", "s3", "cp", f"s3://madewithml/{github_username}/mlflow/", str(MODEL_REGISTRY), "--recursive"])