Improve error message for missing return

[pingsutw]

Tracking issue

NA

Why are the changes needed?

What changes were proposed in this pull request?

How was this patch tested?

import os

from click.testing import CliRunner

from flytekit import task, workflow, ImageSpec
from flytekit.clis.sdk_in_container import pyflyte

default = ["pandas", "mypy"]

image = ImageSpec(registry="pingsutw", packages=[*default])


@task(container_image=image)
def t1() -> int:
    return 3 + 2


@task(container_image=os.getenv("IMAGE"))
def t2(a: int) -> int:
    return a + 3


@workflow()
def wf(a: int = 3) -> int:
    t1()
    t2(a=a)


if __name__ == '__main__':
    runner = CliRunner()
    result = runner.invoke(
        pyflyte.main,
        [
            "-vv",
            "register",
            "/Users/kevin/git/flytekit/flyte-example/improve_image_spec.py",
        ]
    )
    print(result.stdout)

Setup process

Screenshots

Before:

After:

Check all the applicable boxes

• I updated the documentation accordingly.
• All new and existing tests passed.
• All commits are signed-off.

Related PRs

NA

Docs link

NA

[codecov]

Codecov Report

Attention: Patch coverage is 41.66667% with 21 lines in your changes missing coverage. Please review.

Project coverage is 41.64%. Comparing base (993ff59) to head (13d7793).
Report is 4 commits behind head on master.

Files	Patch %	Lines
flytekit/core/utils.py	30.00%	7 Missing ⚠️
flytekit/exceptions/utils.py	0.00%	6 Missing ⚠️
flytekit/core/interface.py	42.85%	4 Missing ⚠️
flytekit/exceptions/user.py	69.23%	4 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##           master    #2551      +/-   ##
==========================================
- Coverage   41.67%   41.64%   -0.03%     
==========================================
  Files         184      184              
  Lines       18784    18802      +18     
  Branches     3926     3708     -218     
==========================================
+ Hits         7828     7830       +2     
- Misses      10824    10863      +39     
+ Partials      132      109      -23     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[Future-Outlier]

LGTM