From ce3e2324502348491824a53f0f382f6de2a077af Mon Sep 17 00:00:00 2001
From: davidmirror-ops <david espejo@union.ai>
Date: Wed, 15 Nov 2023 05:49:45 -0500
Subject: [PATCH] Add missing imports to map_task example

Signed-off-by: davidmirror-ops <david espejo@union.ai>
---
 docs/getting_started/optimizing_tasks.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/getting_started/optimizing_tasks.md b/docs/getting_started/optimizing_tasks.md
index 6a2632c74..17cf07ec9 100644
--- a/docs/getting_started/optimizing_tasks.md
+++ b/docs/getting_started/optimizing_tasks.md
@@ -110,9 +110,9 @@ scale our workload:
 
 ```{code-cell} ipython3
 import math
-from typing import Tuple
+from typing import Tuple, List
 
-from flytekit import map_task
+from flytekit import map_task, task, workflow
 
 
 @task