fix: use any for n_epochs (#499)

* fix: use custom marshaler for n_epochs * chore: use any for n_epochs
2023-10-06 11:32:21 +02:00
parent 84f77a0acd
commit 533935e4fc
2 changed files with 18 additions and 3 deletions
--- a/fine_tuning_job.go
+++ b/fine_tuning_job.go
@@ -24,7 +24,7 @@ type FineTuningJob struct {
 }

 type Hyperparameters struct {
-	Epochs int `json:"n_epochs"`
+	Epochs any `json:"n_epochs,omitempty"`
 }

 type FineTuningJobRequest struct {