fix: minor type for chat and rerank create_finetuned_model example (#121

) Co-authored-by: CoderHam <[email protected]>
cohere-ai · Sep 12, 2024 · c265a25 · c265a25
1 parent 9572275
commit c265a25
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/fern/pages/fine-tuning/chat-fine-tuning/chat-starting-the-training.mdx b/fern/pages/fine-tuning/chat-fine-tuning/chat-starting-the-training.mdx
@@ -240,7 +240,7 @@ create_response = co.finetuning.create_finetuned_model(
       base_model=BaseModel(
         base_type="BASE_TYPE_CHAT",
       ),
-      dataset_id=my-chat_dataset.id,
+      dataset_id=chat_dataset.id,
       hyperparameters=hp,
       wandb=wnb_config,
     ),

diff --git a/fern/pages/fine-tuning/rerank-fine-tuning/rerank-starting-the-training.mdx b/fern/pages/fine-tuning/rerank-fine-tuning/rerank-starting-the-training.mdx
@@ -102,7 +102,7 @@ create_response = co.finetuning.create_finetuned_model(
         name="english",
         base_type="BASE_TYPE_RERANK",
       ),
-      dataset_id=my-rerank_dataset.id,
+      dataset_id=rerank_dataset.id,
     ),
   )
 )