Add multi file error aggregation strategy

flyteorg · Oct 9, 2024 · 2df9e4d · 2df9e4d
1 parent 8f4629f
commit 2df9e4d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/flyteplugins/go/tasks/plugins/k8s/kfoperators/pytorch/pytorch.go b/flyteplugins/go/tasks/plugins/k8s/kfoperators/pytorch/pytorch.go
@@ -28,7 +28,9 @@ type pytorchOperatorResourceHandler struct {
 var _ k8s.Plugin = pytorchOperatorResourceHandler{}
 
 func (pytorchOperatorResourceHandler) GetProperties() k8s.PluginProperties {
-	return k8s.PluginProperties{}
+	return k8s.PluginProperties{
+		ErrorAggregationStrategy: k8s.EarliestErrorAggregationStrategy,
+	}
 }
 
 // Defines a func to create a query object (typically just object and type meta portions) that's used to query k8s

diff --git a/flyteplugins/go/tasks/plugins/k8s/kfoperators/pytorch/pytorch_test.go b/flyteplugins/go/tasks/plugins/k8s/kfoperators/pytorch/pytorch_test.go
@@ -712,7 +712,9 @@ func TestGetLogsElastic(t *testing.T) {
 
 func TestGetProperties(t *testing.T) {
 	pytorchResourceHandler := pytorchOperatorResourceHandler{}
-	expected := k8s.PluginProperties{}
+	expected := k8s.PluginProperties{
+		ErrorAggregationStrategy: k8s.EarliestErrorAggregationStrategy,
+	}
 	assert.Equal(t, expected, pytorchResourceHandler.GetProperties())
 }