diff --git a/openapi.yaml b/openapi.yaml
index 4821b0f..348c61b 100644
--- a/openapi.yaml
+++ b/openapi.yaml
@@ -2599,6 +2599,9 @@ paths:
                 wandb_name:
                   type: string
                   description: The Weights & Biases name for your run.
+                wandb_entity:
+                  type: string
+                  description: The Weights & Biases entity for your run.
                 train_on_inputs:
                   oneOf:
                     - type: boolean