#1690 Fix task type version in pytorch and mpi task plugin
Pull request opened by
yubofredwang
TL;DR
Previously the pytorch plugin is missing comments. We would suggest user to not set configurations for individual replica groups as they should essentially be the same for worker and master. The only fields user should set is number of worker and restart policies.
Fix a bug in pytorch and mpijob to use task type version 1.
Should be merged in after:
flyteorg/flyteplugins#359
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
Smoke tested on the following with corrected backend plugin
1. Elastic Pytorch Job
2. MPI Task
3. Horovod Job
flyteorg/flytekit
✅ All checks have passed
30/30 successful checks