Hi Flyte team - I was trying to write some Flyte doctests using

sphinx.ext.doctest

today and had some trouble with compatibility of doctest and the

task

decorators - put in a small fix and was hoping to get 👀 on it! https://github.com/flyteorg/flytekit/pull/873

Cc @jeev @Yee