I've pushed the mentioned changes to https://github.com/flyteorg/flytekit/pull/1269/files @Samhita Alla

Thanks, @cryptic! Will review the changes and get back to you.

I don't have any checks (0 checks) on my PR, is something wrong here?

Um yeah; I see that too! Can you do a force push?

Okay, so i just updated my fork and it fixed it

Hey. We shouldn’t be using

tensor_dtype

as class variable in this case because it needs to be set a different value within

to_literal

method and that won’t be known to

to_python_value

method. If you want to work on this, you’ll need to send

tensor_dtype

as

format

of the blob.

I've made the suggested changes, please have a look

Thanks!

I hope the changes are correct?

LGTM! I’ll merge it soon.