acoustic-carpenter-78188
10/21/2025, 8:23 PMpytorch.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:
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/flyteacoustic-carpenter-78188
10/23/2025, 2:30 AM