From a2e628fd80ab395a285be5f634a8993b45b544f0 Mon Sep 17 00:00:00 2001
From: Shuying Liang <shuying.liang@gmail.com>
Date: Fri, 10 Jan 2025 19:38:53 -0800
Subject: [PATCH] remove unnecessary line and move operations files to flyte
 repo

---
 .../source/plugins/k8sstatefuldataservice.rst | 12 +++
 .../k8sdataservice/__init__.py                |  1 -
 .../k8s_ops/k8s-service-agent-role.yaml       | 86 -------------------
 .../k8s-service-agent-rolebinding.yaml        | 17 ----
 4 files changed, 12 insertions(+), 104 deletions(-)
 create mode 100644 docs/source/plugins/k8sstatefuldataservice.rst
 delete mode 100644 plugins/flytekit-k8sdataservice/k8s_ops/k8s-service-agent-role.yaml
 delete mode 100644 plugins/flytekit-k8sdataservice/k8s_ops/k8s-service-agent-rolebinding.yaml

diff --git a/docs/source/plugins/k8sstatefuldataservice.rst b/docs/source/plugins/k8sstatefuldataservice.rst
new file mode 100644
index 0000000000..09abe42d29
--- /dev/null
+++ b/docs/source/plugins/k8sstatefuldataservice.rst
@@ -0,0 +1,12 @@
+.. k8sstatefuldataservice:
+
+###################################################
+Kubernetes StatefulSet Data Service API reference
+###################################################
+
+.. tags:: Integration, DeepLearning, MachineLearning, Kubernetes, GNN
+
+.. automodule:: flytekitplugins.k8sdataservice
+   :no-members:
+   :no-inherited-members:
+   :no-special-members:
diff --git a/plugins/flytekit-k8sdataservice/flytekitplugins/k8sdataservice/__init__.py b/plugins/flytekit-k8sdataservice/flytekitplugins/k8sdataservice/__init__.py
index d9238de5c5..f85aa90be0 100644
--- a/plugins/flytekit-k8sdataservice/flytekitplugins/k8sdataservice/__init__.py
+++ b/plugins/flytekit-k8sdataservice/flytekitplugins/k8sdataservice/__init__.py
@@ -13,4 +13,3 @@
 from .agent import DataServiceAgent  # noqa: F401
 from .sensor import CleanupSensor  # noqa: F401
 from .task import DataServiceConfig, DataServiceTask  # noqa: F401
-# from .dataservice_sensor_engine import DSSensorEngine  # noqa: F401
diff --git a/plugins/flytekit-k8sdataservice/k8s_ops/k8s-service-agent-role.yaml b/plugins/flytekit-k8sdataservice/k8s_ops/k8s-service-agent-role.yaml
deleted file mode 100644
index 4236f533f7..0000000000
--- a/plugins/flytekit-k8sdataservice/k8s_ops/k8s-service-agent-role.yaml
+++ /dev/null
@@ -1,86 +0,0 @@
-# Example of the role/binding set up for the data service to create/update/delete resources in the sandbox flyte namespace
-apiVersion: rbac.authorization.k8s.io/v1
-kind: Role
-metadata:
-  name: flyte-flyteagent-role
-  namespace: flyte
-  labels:
-    app.kubernetes.io/name: flyteagent
-    app.kubernetes.io/instance: flyte
-rules:
-- apiGroups:
-      - apps
-  resources:
-    - statefulsets
-    - statefulsets/status
-    - statefulsets/scale
-    - statefulsets/finalizers
-  verbs:
-    - get
-    - list
-    - watch
-    - create
-    - update
-    - delete
-    - patch
-- apiGroups:
-  - ""
-  resources:
-  - pods
-  - configmaps
-  - serviceaccounts
-  - secrets
-  - pods/exec
-  - pods/log
-  - pods/status
-  - services
-  verbs:
-  - '*'
-# Allow Event recording access
-- apiGroups:
-  - ""
-  resources:
-  - events
-  verbs:
-  - create
-  - update
-  - delete
-  - patch
-# Allow Access to Podtemplates
-- apiGroups:
-  - ""
-  resources:
-  - podtemplates
-  verbs:
-  - get
-  - list
-  - watch
-# Allow Access to all resources under flyte.lyft.com
-- apiGroups:
-  - flyte.lyft.com
-  resources:
-  - flyteworkflows
-  - flyteworkflows/finalizers
-  verbs:
-  - get
-  - list
-  - watch
-  - create
-  - update
-  - delete
-  - patch
-  - post
-  - deletecollection
-# Allow Access to plugin objects.
-- apiGroups:
-  - 'kubeflow.org'
-  resources:
-  - "*"
-  verbs:
-  - get
-  - list
-  - watch
-  - create
-  - update
-  - delete
-  - patch
diff --git a/plugins/flytekit-k8sdataservice/k8s_ops/k8s-service-agent-rolebinding.yaml b/plugins/flytekit-k8sdataservice/k8s_ops/k8s-service-agent-rolebinding.yaml
deleted file mode 100644
index 183d0c6a22..0000000000
--- a/plugins/flytekit-k8sdataservice/k8s_ops/k8s-service-agent-rolebinding.yaml
+++ /dev/null
@@ -1,17 +0,0 @@
-# Example of the role/binding set up for the data service to create/update/delete resources in the sandbox flyte namespace
-apiVersion: rbac.authorization.k8s.io/v1
-kind: RoleBinding
-metadata:
-  name: flyte-flyteagent-rolebinding
-  namespace: flyte
-  labels:
-    app.kubernetes.io/name: flyteagent
-    app.kubernetes.io/instance: flyte
-roleRef:
-  apiGroup: rbac.authorization.k8s.io
-  kind: Role
-  name: flyte-flyteagent-role
-subjects:
-- kind: ServiceAccount
-  name: flyteagent
-  namespace: flyte