fix: add ContainerTask to type hint

-e
Signed-off-by: machichima <[email protected]>

Author: machichima

flytekit/core/array_node_map_task.py

@@ -98,7 +98,7 @@ def __init__(
             actual_task.python_interface, self._bound_inputs, set(), output_as_list_of_optionals
         )
 
-        self._run_task: Union[PythonFunctionTask, PythonInstanceTask] = actual_task  # type: ignore
+        self._run_task: Union[PythonFunctionTask, PythonInstanceTask, ContainerTask] = actual_task  # type: ignore
         if isinstance(actual_task, PythonInstanceTask):
             mod = actual_task.task_type
             f = actual_task.lhs
@@ -173,7 +173,7 @@ def concurrency(self) -> Optional[int]:
         return self._concurrency
 
     @property
-    def python_function_task(self) -> Union[PythonFunctionTask, PythonInstanceTask]:
+    def python_function_task(self) -> Union[PythonFunctionTask, PythonInstanceTask, ContainerTask]:
         return self._run_task
 
     @property