#6156 [BUG] [flytekit] Task with torch.Tensor input fails when executing with flytekit remote access Issue created by pimdh Describe the bug Consider following task definition: @task def flyte_test_task(x: torch.Tensor) -> None: print(x) Then register. Then execute with: PROJECT = ... DOMAIN = ... remote = FlyteRemote(config=Config.auto()) task = remote.fetch_task(name="flyte_test_task", project=PROJECT, domain=DOMAIN) execution = remote.execute( task, project=PROJECT, domain=DOMAIN, inputs={"x": torch.tensor([1, 2, 3])}, ) This triggers error:

TypeError: No automatic conversion found from type <class 'torch.Tensor'> to FlyteFile.Supported (os.PathLike, str, Flytefile)

Expected behavior The execution should be created successfully. Additional context to reproduce Using flytekit version

1.14.3

Screenshots No response Are you sure this issue hasn't been raised already? • Yes Have you read the Code of Conduct? • Yes flyteorg/flyte