<#6689 Pytorch proto v2 migration> Pull request op...
# flyte-github
a
#6689 Pytorch proto v2 migration Pull request opened by WangWang0226 ### Why are the changes needed?
pytorch.proto
existed only in v1 (
flyteidl/plugins
) and was missing from v2 (
flyteidl2/plugins
). Without this file, the PyTorch plugin messages were not available in the v2 IDL and downstream codegen. Additionally, the comment in the proto referenced a dead link to the Kubeflow Training Operator types. --- ### What changes were proposed in this pull request? • Migrate v1 → v2: Moved the v1
pytorch.proto
content into the v2 plugin folder:
flyteidl2/plugins/pytorch.proto
. • Update Go package path: Adjusted
option go_package = "<http://github.com/flyteorg/flyte/v2/gen/go/flyteidl2/plugins|github.com/flyteorg/flyte/v2/gen/go/flyteidl2/plugins>";
to match the v2 Go module import path. • Fix broken reference link in comments: Replaced the stale link with a stable permalink to the Kubeflow Training Operator v1 PyTorch types:
<https://github.com/kubeflow/trainer/blob/e31d11faa9f6ce5111b60c01079d39295589e0ef/pkg/apis/kubeflow.org/v1/pytorch_types.go>
--- ### How was this patch tested? • Ran
make gen
to trigger Buf-based code generation. • Verified the generated files are emitted under the right packages. • Confirmed successful compilation and correct import paths in the generated code. --- ### Labels
changed
--- ### Setup process No special setup. Use existing targets:
Copy code
make gen
--- ### Screenshots N/A (proto-only change). --- ### Check all the applicable boxes • I updated the documentation accordingly (proto comment link fixed). • All new and existing tests passed. (No runtime logic changed; proto-only codegen verified.) • All commits are signed-off. --- ### Related PRs N/A (v1 → v2 migration within this repo). --- ### Docs link N/A (internal proto migration; inline comment now points to a stable upstream reference). flyteorg/flyte