diff --git a/hack/swagger/main.go b/hack/swagger/main.go
index 491aeec36f..c2ff88c641 100644
--- a/hack/swagger/main.go
+++ b/hack/swagger/main.go
@@ -62,9 +62,6 @@ func main() {
 	defs := spec.Definitions{}
 
 	refCallback := func(name string) spec.Ref {
-		// return spec.MustCreateRef("#/definitions/" + common.EscapeJsonPointer(swaggify(name, k8sVersion, jobSetVersion)))
-		// return spec.MustCreateRef(getRefDefinition(name, k8sVersion, jobSetVersion))
-
 		if strings.HasPrefix(name, "k8s.io") {
 			return spec.MustCreateRef(k8sOpenAPISpec + "#/definitions/" + swaggify(name))
 		} else if strings.HasPrefix(name, "sigs.k8s.io/jobset") {
@@ -103,7 +100,6 @@ func main() {
 func swaggify(name string) string {
 	name = strings.Replace(name, "github.com/kubeflow/trainer/pkg/apis/", "", -1)
 	name = strings.Replace(name, "sigs.k8s.io/jobset/api/", "", -1)
-
 	name = strings.Replace(name, "k8s.io", "io.k8s", -1)
 	name = strings.Replace(name, "/", ".", -1)
 	return name
diff --git a/sdk/kubeflow/trainer/api/trainer_client.py b/sdk/kubeflow/trainer/api/trainer_client.py
index ee79faf0d4..747c639588 100644
--- a/sdk/kubeflow/trainer/api/trainer_client.py
+++ b/sdk/kubeflow/trainer/api/trainer_client.py
@@ -72,7 +72,6 @@ def __init__(
         k8s_client = client.ApiClient(client_configuration)
         self.custom_api = client.CustomObjectsApi(k8s_client)
         self.core_api = client.CoreV1Api(k8s_client)
-        # self.api_client = ApiClient()
 
         self.namespace = namespace