#1507 Fix pytorch transformer
Pull request opened by
samhita-alla
Signed-off-by: Samhita Alla
aallasamhita@gmail.com
TL;DR
This PR fixes type fetch for torch types.
Type
☑︎ Bug Fix
☐ Feature
☐ Plugin
Are all requirements met?
☐ Code completed
☐ Smoke tested
☐ Unit tests added
☐ Code documentation added
☐ Any pending items have an associated Issue
Complete description
The guess_python_type method has been moved to the subclasses without retaining it in the parent class. This will return the correct type when fetching the execution output via FlyteRemote.
Tracking Issue
Fixes
flyteorg/flyte#3148
Follow-up issue
NA
OR
https://github.com/flyteorg/flyte/issues/
flyteorg/flytekit
Codecov: 46.15% of diff hit (target 69.45%)
✅ 29 other checks have passed
29/30 successful checks