Hi, I have a simple question: From the Flyte documentation and tutorial, it seems classes are not su...

sparse-family-37927

03/17/2023, 7:41 PM

Hi, I have a simple question: From the Flyte documentation and tutorial, it seems classes are not supported by Flyte. Is this correct? If not, is there any material that you can share about using classes in Flyte. Thanks!

glamorous-carpet-83516

03/17/2023, 8:23 PM

you means using classes as task input / output ?

sparse-family-37927

03/17/2023, 8:33 PM

Like can we define some tasks, workflows, launchplans within a class.

sparse-family-37927

03/17/2023, 8:34 PM

Or in another way, the questions is: can we convert the functions in classes to tasks, workflows and launchplans.

glamorous-carpet-83516

03/17/2023, 8:37 PM

why do you want use task or workflow in the class?

sparse-family-37927

03/17/2023, 8:37 PM

I am asking this question as we wanted to build a flyte pipeline in my company and we noticed that there are lots of classes in this project. If we need to convert these classes to be functions (which can be decorated to be tasks), then there will be a lot of work.

sparse-family-37927

03/17/2023, 8:38 PM

So do we have any workaround in this situation?

glamorous-carpet-83516

03/17/2023, 8:39 PM

do you mind share some examples of your classes? I’m thinking how to easily convert your class to flyte task.

sparse-family-37927

03/17/2023, 8:41 PM

I may need to ask in my team to see if it is good to share some examples. I will let you know later. Thanks anyway. This is helpful!

glamorous-carpet-83516

03/17/2023, 8:46 PM

I wrote a simple example, is this what you want?

Copy code

from flytekit import workflow, task


class MyClass:

    @task
    def t1(self):
        return 'hello world'


@workflow
def wf():
    MyClass().t1()

sparse-family-37927

03/17/2023, 8:49 PM

yes, does flyte support this? My understanding is no. But I am not sure.

sparse-family-37927

03/17/2023, 8:50 PM

In some cases, we may also want to have

Copy code

from flytekit import workflow, task


class MyClass:

    @task
    def t1(self):
        return 'hello world'

    @task
    def t2(self):
        return 'hello world too'


@workflow
def wf():
    MyClass().t1()
    MyClass().t2()

glamorous-carpet-83516

03/17/2023, 8:53 PM

Not now, but I think it’s doable. mind creating an issue here [flyte-core]. we will work on that. contributions welcome too.

glamorous-carpet-83516

03/17/2023, 8:53 PM

[flyte-core]

user

03/17/2023, 8:53 PM

⭐ Create a new Flyte Core Feature issue: https://github.com/flyteorg/flyte/issues/new?assignees=&labels=enhancement%2Cuntriaged&template=feature_request.yaml&title=%5BCore+feature%5D+

sparse-family-37927

03/17/2023, 9:10 PM

just created an issue for this. Not sure if what I filled in is correct or not. Feel free to add anything to make it more clear to people. Thanks!