update test

Signed-off-by: Niels Bantilan <niels.bantilan@gmail.com>

Author: cosmicBboy

flytekit/models/task.py

@@ -701,7 +701,7 @@ def from_flyte_idl(cls, pb2_object):
 
 
 class Task(_common.FlyteIdlEntity):
-    def __init__(self, id, closure, short_description):
+    def __init__(self, id, closure, short_description=None):
         """
         :param flytekit.models.core.identifier.Identifier id: The (project, domain, name) identifier for this task.
         :param TaskClosure closure: The closure for the underlying workload.
@@ -729,6 +729,7 @@ def closure(self):
     @property
     def short_description(self):
         """
+        The short description of the task.
         :rtype: str
         """
         return self._short_description

tests/flytekit/unit/models/test_tasks.py

@@ -299,6 +299,7 @@ def test_task(task_closure):
     obj = task.Task(
         identifier.Identifier(identifier.ResourceType.TASK, "project", "domain", "name", "version"),
         task_closure,
+        "my short description",
     )
     assert obj.id.project == "project"
     assert obj.id.domain == "domain"