support fast register

Signed-off-by: Kevin Su <[email protected]>

flytekit/tools/translator.py

@@ -180,7 +180,10 @@ def get_serializable_task(
 
     if settings.should_fast_serialize():
         # This handles container tasks.
-        if container and isinstance(entity, (PythonAutoContainerTask, MapPythonTask)):
+        if container and (
+            isinstance(entity, PythonAutoContainerTask)
+            or (isinstance(entity, MapPythonTask) and isinstance(entity.run_task, PythonAutoContainerTask))
+        ):
             # For fast registration, we'll need to muck with the command, but on
             # ly for certain kinds of tasks. Specifically,
             # tasks that rely on user code defined in the container. This should be encapsulated by the auto container

tests/flytekit/unit/core/test_map_task.py

@@ -120,6 +120,7 @@ def test_serialization_with_raw_container(serialization_settings):
     assert task_spec.template.custom["minSuccessRatio"] == 1.0
     assert task_spec.template.type == "container_array"
     assert task_spec.template.task_type_version == 1
+    assert task_spec.template.container.args is None
     assert task_spec.template.container.command == [
         "python",
         "test.py",