https://flyte.org
Join Slack
Menu
Home
Archive
Hi flyte community! I have a question about the resolver path. I noticed that if I use `pyflyte regi...
# flyte-support
t
Hi flyte community! I have a question about the resolver path. I noticed that if I use 
pyflyte register
to register the workflow, the resolver path would be 
app.<…>.default_task_resolver
, but if programmatically register workflow using 
register_workflow()
, the resolver path would be 
module.<…>.default_task_resolver
(
module
being the root of the workflow code). Is there a way to change the resolver path back to 
app
if I use 
register_workflow
?
t
can you elaborate a bit more? what is this default_task_resolver?
can you maybe copy paste in a 
tree
structure and some sample repro code?
t
Sorry for the late reply I was distracted yesterday. To be more specific, if I register programmatically using 
register_workflow
, and the resolver path would be 
run_my_workflow.pip_pypi__flytekit.flytekit.core.python_auto_container.default_task_resolver
. And 
run_my_workflow
is the parent directory bazel target name of the script (same as the script name in this case) that calls 
register_workflow
And if I register my workflow using CLI 
pyflyte register
, the resolver path in the task would look like 
app.pip_pypi__flytekit.flytekit.core.python_auto_container.default_task_resolver
So I wonder if I can set the root of the resolver path to 
app
when calling 
register_workflow
I will come up with a repro in a bit. Thanks for the help in advance!
t
also are you using conda or pip?
i’m not too familiar with conda
t
pip, but actually, it’s bazel-pip
I think I figured out how to override the default resolver, just need to override the 
location
in the 
DefaultTaskResolver
Here is my solution:
Copy code
TASK_RESOLVER_LOCATION = "app.pip_pypi__flytekit.flytekit.core.python_auto_container.default_task_resolver"
class CustomTaskResolver(DefaultTaskResolver):
    def __init__(self, *args, resolver_location = TASK_RESOLVER_LOCATION, **kwargs):
        super().__init__(*args, **kwargs)
        self.resolver_location = resolver_location

    @property
    def location(self) -> str:
        return self.resolver_location
    

@task(task_resolver=CustomTaskResolver())
def my_task(a: int, b: int) -> int:
    return a + b


@task(task_resolver=CustomTaskResolver())
def my_other_task(m: int, n: int) -> int:
    return 2 * (m - n)
152 Views
Open in Slack
PreviousNext
Powered by

Flyte

Flyte enables production-grade orchestration for machine learning workflows and data processing created to accelerate local workflows to production.

Powered by