Update sdk/python/v1beta1/kubeflow/katib/api/katib_client.py

Co-authored-by: Andrey Velichkevich <andrey.velichkevich@gmail.com>
kubeflow · Aug 4, 2023 · ada3ec9 · ada3ec9
1 parent abd614d
commit ada3ec9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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.