diff --git a/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py b/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py
index 7ef770cc39c..a623d9956ed 100644
--- a/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py
+++ b/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py
@@ -194,6 +194,7 @@ def tune(
             For example:
                 {
                     "cpu": "1",
+                    "gpu": "1",
                     "memory": "2Gi",
                 }
             This parameter is optional and defaults to None.