WandB API Param Name fix (#102)

* Wandb api key fix

* v 1.1.2

---------

Co-authored-by: Arsh Zahed <[email protected]>

Author: azahed98

pyproject.toml

@@ -12,7 +12,7 @@ build-backend = "poetry.masonry.api"
 
 [tool.poetry]
 name = "together"
-version = "1.1.1"
+version = "1.1.2"
 authors = [
     "Together AI <[email protected]>"
 ]

src/together/resources/finetune.py

@@ -66,7 +66,7 @@ def create(
             batch_size=batch_size,
             learning_rate=learning_rate,
             suffix=suffix,
-            wandb_api_key=wandb_api_key,
+            wandb_key=wandb_api_key,
         ).model_dump()
 
         response, _, _ = requestor.request(
@@ -280,7 +280,7 @@ async def create(
             batch_size=batch_size,
             learning_rate=learning_rate,
             suffix=suffix,
-            wandb_api_key=wandb_api_key,
+            wandb_key=wandb_api_key,
         ).model_dump()
 
         response, _, _ = await requestor.arequest(

src/together/types/finetune.py

@@ -118,7 +118,7 @@ class FinetuneRequest(BaseModel):
     # up to 40 character suffix for output model name
     suffix: str | None = None
     # weights & biases api key
-    wandb_api_key: str | None = None
+    wandb_key: str | None = None
 
 
 class FinetuneResponse(BaseModel):