Hello all I m having an issue when trying to use reference t

Question

Hello all I m having an issue when trying to use reference tasks When building image I m getting this error ` ReferenceTask object has no attribute task function ` Can someone help me I would be very appreciated

Victor Gustavo da Silva Oliveira · Answer

I m using some tasks to test This is an example ```from flytekit import task task def task example n int gt int Simple example task for library building Parameters n int name of the parameter for the task will be derived from the name of the input variable the type will be automatically deduced to be Types Integer Return int The label for the output will be automatically assigned and type will be deduced from the annotation return n n``` And I m referencing like this ```from flytekit import reference task reference task project= flyte tasks domain= development name= src flyte tasks example task example task task example def task example n int gt int ```

Ketan (kumare3) · Answer

Seems odd do you have bigger stack trace

Victor Gustavo da Silva Oliveira · Answer

I think we understood what was the problem We are using a custom class as return in one of the referenced tasks But I thought that it wont be needed to recreate in other project if we are using reference class

Victor Gustavo da Silva Oliveira · Answer

Even when we added the custom class the error keeps happening

Samhita Alla · Answer

< Victor Gustavo da Silva Oliveira> if you could share the actual code snippet and stack trace that d be helpful to find the source of the error

Victor Gustavo da Silva Oliveira · Answer

Sorry it took so long It is happening in a colleague s task Here it is ```Traceback most recent call last File home jbudke miniforge3 envs dedup bin pyflyte line 8 in lt module gt sys exit main File home jbudke local lib python3 8 site packages click core py line 1130 in call return self main args kwargs File home jbudke local lib python3 8 site packages click core py line 1055 in main rv = self invoke ctx File home jbudke local lib python3 8 site packages click core py line 1657 in invoke return process result sub ctx command invoke sub ctx File home jbudke local lib python3 8 site packages click core py line 1657 in invoke return process result sub ctx command invoke sub ctx File home jbudke local lib python3 8 site packages click core py line 1651 in invoke cmd name cmd args = self resolve command ctx args File home jbudke local lib python3 8 site packages click core py line 1698 in resolve command cmd = self get command ctx cmd name File home jbudke miniforge3 envs dedup lib python3 8 site packages flytekit clis sdk in container run py line 607 in get command entity = load naive entity module exe entity project root File home jbudke miniforge3 envs dedup lib python3 8 site packages flytekit clis sdk in container run py line 447 in load naive entity importlib import module module name File home jbudke miniforge3 envs dedup lib python3 8 importlib init py line 127 in import module return bootstrap gcd import name level package level File lt frozen importlib bootstrap gt line 1014 in gcd import File lt frozen importlib bootstrap gt line 991 in find and load File lt frozen importlib bootstrap gt line 975 in find and load unlocked File lt frozen importlib bootstrap gt line 671 in load unlocked File lt frozen importlib bootstrap external gt line 843 in exec module File lt frozen importlib bootstrap gt line 219 in call with frames removed File home jbudke Área de Trabalho image intelligence dsc image intelligence src flyte tasks workflows image workflow py line 13 in lt module gt def image workflow File home jbudke miniforge3 envs dedup lib python3 8 site packages flytekit core workflow py line 739 in workflow return wrapper workflow function File home jbudke miniforge3 envs dedup lib python3 8 site packages flytekit core workflow py line 734 in wrapper workflow instance compile File home jbudke miniforge3 envs dedup lib python3 8 site packages flytekit core workflow py line 614 in compile workflow outputs = exception scopes user entry point self workflow function input kwargs File home jbudke miniforge3 envs dedup lib python3 8 site packages flytekit exceptions scopes py line 198 in user entry point return wrapped args kwargs File home jbudke Área de Trabalho image intelligence dsc image intelligence src flyte tasks workflows image workflow py line 41 in image workflow all samples = map task File home jbudke miniforge3 envs dedup lib python3 8 site packages flytekit core map task py line 274 in map task return MapPythonTask task function concurrency=concurrency min success ratio=min success ratio kwargs File home jbudke miniforge3 envs dedup lib python3 8 site packages flytekit core tracker py line 35 in call o = super InstanceTrackingMeta cls call args kwargs File home jbudke miniforge3 envs dedup lib python3 8 site packages flytekit core map task py line 58 in init mod f = tracker extract task module python function task task function File home jbudke miniforge3 envs dedup lib python3 8 site packages flytekit core python function task py line 156 in task function return self task function AttributeError ReferenceTask object has no attribute task function ```

Samhita Alla · Answer

Are you using the referenced task in a map task

Samhita Alla · Answer

Is it possible for you to share a redacted code snippet

Victor Gustavo da Silva Oliveira · Answer

Yes the idea is to use the referenced task as a mapped task due to the size of the dataset it handles

Victor Gustavo da Silva Oliveira · Answer

I ll try to update the code as soon as I can

Victor Gustavo da Silva Oliveira · Answer

But the general idea is to use the referenced task which uses a model to do some image predictions in a map task for better performance

Samhita Alla · Answer

Looks like a bug cc < Eduardo Apolinario eapolinario > Code ```from flytekit import map task reference task workflow reference task project= flytesnacks domain= development name= core flyte basics decorating tasks t1 version= AdGs5FM0Sh2WFM0T6rZemg== def t1 x int gt int workflow def wf x list int = 3 2 return map task t1 x=x ``` Error ```Failed with Unknown Exception lt class AttributeError gt Reason ReferenceTask object has no attribute task function ```

Eduardo Apolinario (eapolinario) · Answer

I agree < Samhita Alla> can you open a gh issue

Samhita Alla · Answer

Ketan (kumare3) · Answer

I do not think this is a bug I think this is a feature request

Ketan (kumare3) · Answer

And this is a hard one