Hello all I'm getting this error when trying to create a task with custom object: ```ValueError: Onl...

dry-ability-69144

03/08/2023, 2:01 PM

Hello all I'm getting this error when trying to create a task with custom object:

Copy code

ValueError: Only generic univariate typing.List[T] type is supported.

I tried to use the example in doc, https://flyte--943.org.readthedocs.build/projects/cookbook/en/943/auto/core/type_system/custom_objects.html

dry-ability-69144

03/08/2023, 3:06 PM

In this case, I adapted the example to create a custom object, and I'm trying to create a map task that receives a list of this objects

high-accountant-32689

03/08/2023, 3:17 PM

Can you share the code, @dry-ability-69144?

dry-ability-69144

03/08/2023, 4:26 PM

The custom class

Copy code

@dataclass_json
@dataclass
class Test_custom_object(object):
    flyte_file: FlyteFile
    test_data: str

A snippet from the class

Copy code

@task(requests=Resources(cpu='1', mem='600Mi'))
def test_task(file: Datum) -> float:
    print(file.flyte_file)

Snippet from workflow

Copy code

all_samples = map_task(
       sample_task,
       metadata=TaskMetadata(retries=1)
    )(file=samples)

map_task(
        test_task,
        metadata=TaskMetadata(retries=1)
    )(file=all_samples)

tall-lock-23197

03/09/2023, 6:00 AM

@dry-ability-69144, which task is throwing the error? And what's the type of

all_samples

list elements?

dry-ability-69144

03/09/2023, 11:18 AM

The error is coming from the latter map task, and

all_samples

is returning me a list of

Test_custom_object

179 Views

Open in Slack

Previous Next

Flyte

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