Fix task type version in pytorch and mpi task plugin

[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

[pingsutw]

LGTM, thank you!

[codecov]

Codecov Report

Merging #1690 (86c93d8) into master (3370a96) will not change coverage.
The diff coverage is n/a.

@@           Coverage Diff           @@
##           master    #1690   +/-   ##
=======================================
  Coverage   71.03%   71.03%           
=======================================
  Files         336      336           
  Lines       30798    30798           
  Branches     5589     5589           
=======================================
  Hits        21876    21876           
  Misses       8375     8375           
  Partials      547      547