chore: use any for n_epochs

sashabaranov · Oct 5, 2023 · 581e933 · 581e933
1 parent 2be7241
commit 581e933
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 39 deletions.
diff --git a/fine_tuning_job.go b/fine_tuning_job.go
@@ -2,7 +2,6 @@ package openai
 
 import (
 	"context"
-	"encoding/json"
 	"fmt"
 	"net/http"
 	"net/url"
@@ -24,40 +23,8 @@ type FineTuningJob struct {
 	TrainedTokens   int             `json:"trained_tokens"`
 }
 
-type HyperparameterNEpochs struct {
-	IntValue    *int    `json:"-"`
-	StringValue *string `json:"-"`
-}
-
-func (h *HyperparameterNEpochs) UnmarshalJSON(data []byte) error {
-	var intValue int
-	var stringValue string
-
-	if err := json.Unmarshal(data, &intValue); err == nil {
-		h.IntValue = &intValue
-		return nil
-	}
-
-	if err := json.Unmarshal(data, &stringValue); err != nil {
-		return err
-	}
-
-	h.StringValue = &stringValue
-	return nil
-}
-
-func (h *HyperparameterNEpochs) MarshalJSON() ([]byte, error) {
-	if h.IntValue != nil {
-		return json.Marshal(*h.IntValue)
-	} else if h.StringValue != nil {
-		return json.Marshal(*h.StringValue)
-	}
-
-	return nil, fmt.Errorf("invalid hyperparameter n_epochs")
-}
-
 type Hyperparameters struct {
-	Epochs *HyperparameterNEpochs `json:"n_epochs,omitempty"`
+	Epochs any `json:"n_epochs,omitempty"`
 }
 
 type FineTuningJobRequest struct {

diff --git a/fine_tuning_job_test.go b/fine_tuning_job_test.go
@@ -21,7 +21,6 @@ func TestFineTuningJob(t *testing.T) {
 	server.RegisterHandler(
 		"/v1/fine_tuning/jobs",
 		func(w http.ResponseWriter, r *http.Request) {
-			nEpochs := "auto"
 			resBytes, _ := json.Marshal(FineTuningJob{
 				Object:         "fine_tuning.job",
 				ID:             testFineTuninigJobID,
@@ -35,10 +34,7 @@ func TestFineTuningJob(t *testing.T) {
 				ValidationFile: "",
 				TrainingFile:   "file-abc123",
 				Hyperparameters: Hyperparameters{
-					Epochs: &HyperparameterNEpochs{
-						IntValue:    nil,
-						StringValue: &nEpochs,
-					},
+					Epochs: "auto",
 				},
 				TrainedTokens: 5768,
 			})