Update vscode decorator

[pingsutw]

Why are the changes needed?

Failed to run the task with vscode decorator locally.

│ ❱ 291 │   │   return wrapper(_task_function)                                 │
│                                                                              │
│ /Users/kevin/git/flytekit/flytekit/core/task.py:269 in wrapper               │
│                                                                              │
│ ❱ 269 │   │   task_instance = TaskPlugins.find_pythontask_plugin(type(task_c │
│                                                                              │
│ /Users/kevin/git/flytekit/flytekit/core/tracker.py:85 in __call__            │
│                                                                              │
│ ❱  85 │   │   o = super(InstanceTrackingMeta, cls).__call__(*args, **kwargs) │
│                                                                              │
│ /Users/kevin/git/flytekit/flytekit/core/python_function_task.py:120 in       │
│ __init__                                                                     │
│                                                                              │
│ ❱ 120 │   │   name, _, _, _ = extract_task_module(task_function)             │
│                                                                              │
│ /Users/kevin/git/flytekit/flytekit/core/tracker.py:327 in                    │
│ extract_task_module                                                          │
│                                                                              │
│ ❱ 327 │   │   inspect_file = inspect.getfile(f)  # type: ignore              │
│                                                                              │
│ /Users/kevin/opt/anaconda3/envs/flytekit-3.10/lib/python3.10/inspect.py:797  │
│ in getfile                                                                   │
│                                                                              │
│ ❱  797 │   raise TypeError('module, class, method, function, traceback, fram │
╰──────────────────────────────────────────────────────────────────────────────╯
TypeError: module, class, method, function, traceback, frame, or code object was
expected, got vscode

What changes were proposed in this pull request?

• Update extract_task_module, so we are able to get the task module and name from ClassDecorator
• Rename _wrap_call to execute
• Add more tests

How was this patch tested?

Run a task with vscode decorator locally

from flytekitplugins.flyin import vscode
from flytekit import task


@task
@vscode(enable=True)
def t1(a: int, b: int) -> int:
    return a // b


if __name__ == '__main__':
    print(t1(a=6, b=3))

Setup process

Screenshots

Check all the applicable boxes

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

[codecov]

Codecov Report

Attention: 4 lines in your changes are missing coverage. Please review.

Comparison is base (47e15b6) 85.98% compared to head (8d5d7cd) 84.87%.
Report is 2 commits behind head on master.

Files	Patch %	Lines
flytekit/core/tracker.py	0.00%	3 Missing ⚠️
...lyin/flytekitplugins/flyin/vscode_lib/decorator.py	83.33%	1 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##           master    #2018      +/-   ##
==========================================
- Coverage   85.98%   84.87%   -1.12%     
==========================================
  Files         308      258      -50     
  Lines       22946    20763    -2183     
  Branches     3468     3475       +7     
==========================================
- Hits        19731    17622    -2109     
+ Misses       2615     2536      -79     
- Partials      600      605       +5     

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

[ByronHsu]

abstractmethod instead of NotImplementedError?

[pingsutw]

done

[eapolinario]

flyin python 3.8 tests are failing: https://github.com/flyteorg/flytekit/actions/runs/7215520972/job/19660153192?pr=2018