[BUG] Flytkit Named Tuples cannot access be accessed if there is only one attribute

[brandon-segal]

Description:

In Flytekit there is no way to mock a named tuple that only has a single member. Mocking a situation like the following results in a AttributeError: 'Promise' object has no attribute 'uri' error.

class GenerateOutputs(NamedTuple):
        uri: str
    
@reference_task(
    ...
)
def generate(
    input:str 
) -> GenerateOutputs:
    ... 
    
@workflow
def my_wf(
   input="default_input"
) -> str:
    output = generate(
      input:str=input
    )
    return output.uri

class TestNamedTuple(TestCase):
    def test_run_workflow(self):
        with task_mock(generate) as fake_generate:
            fake_generate.return_value = GenerateOutputs(uri="test_uri")
            uri = my_wf()
            self.assertIsNotNone(uri)

Ideally we would be able to do something like the following which we can do when the NamedTuple has two attributes:

from typing import NamedTuple
from unittest import TestCase
from flytekit.testing import task_mock
from flytekit import workflow,reference_task

class GenerateOutputs(NamedTuple):
    uri: str
    other_thing:str

@reference_task(
    ...
)
def generate(
    input:str 
) -> GenerateOutputs:
    ... 

@workflow
def my_wf(
    input:str="default_input"
) -> str:
    output = generate(
        input=input
    )
    # Could similarly unpack the tuple and just return the uri 
    return output.uri


class TestNamedTuple(TestCase):
    def test_run_workflow(self):
        with task_mock(generate) as fake_generate:
            fake_generate.return_value = GenerateOutputs(uri="test_uri",other_thing="test")
            uri = my_wf()
            self.assertIsNotNone(uri)

[welcome]

Thank you for opening your first issue here! 🛠

[github-actions]

Hello 👋, This issue has been inactive for over 9 months. To help maintain a clean and focused backlog, we'll be marking this issue as stale and will close the issue if we detect no activity in the next 7 days. Thank you for your contribution and understanding! 🙏

[wild-endeavor]

Something similar to this will soon be possible. We still haven't figured out the namedtuple issue yet, but you should be able to bind to dataclass components soon. #4232

Please re-open if this is not sufficient.