Skip to content

Commit

Permalink
Moving all plugins to a common package, for easy loading (flyteorg#423)
Browse files Browse the repository at this point in the history
* Moving all plugins to a common package, for easy loading

Signed-off-by: Ketan Umare <[email protected]>

* updated

Signed-off-by: Ketan Umare <[email protected]>
  • Loading branch information
kumare3 authored Apr 12, 2022
1 parent 027fedb commit 3d48153
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 11 deletions.
12 changes: 1 addition & 11 deletions flytepropeller/cmd/controller/main.go
Original file line number Diff line number Diff line change
@@ -1,17 +1,7 @@
package main

import (
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/array/awsbatch"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/array/k8s"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/hive"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/kfoperators/mpi"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/kfoperators/pytorch"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/pod"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/sagemaker"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/spark"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/webapi/athena"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/webapi/bigquery"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/webapi/snowflake"
_ "github.com/flyteorg/flytepropeller/plugins"
"github.com/flyteorg/flytestdlib/contextutils"
"github.com/flyteorg/flytestdlib/promutils/labeled"

Expand Down
17 changes: 17 additions & 0 deletions flytepropeller/plugins/loader.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
// Package plugins facilitates all the plugins that should be loaded by FlytePropeller
package plugins

import (
// Common place to import all plugins, so that it can be imported by Singlebinary (flytelite) or by propeller main
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/array/awsbatch"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/array/k8s"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/hive"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/kfoperators/mpi"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/kfoperators/pytorch"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/pod"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/sagemaker"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/k8s/spark"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/webapi/athena"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/webapi/bigquery"
_ "github.com/flyteorg/flyteplugins/go/tasks/plugins/webapi/snowflake"
)

0 comments on commit 3d48153

Please sign in to comment.