Rename nEpochs to n_epochs in FineTuneRequest (TheoKanning#148)

* Rename nEpochs to n_epochs in FineTuneRequest * Remove unused import
arqu-co · Mar 8, 2023 · 27c9089 · 27c9089
1 parent ec701cd
commit 27c9089
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/api/src/main/java/com/theokanning/openai/finetune/FineTuneRequest.java b/api/src/main/java/com/theokanning/openai/finetune/FineTuneRequest.java
@@ -37,6 +37,7 @@ public class FineTuneRequest {
     /**
      * The number of epochs to train the model for. An epoch refers to one full cycle through the training dataset.
      */
+    @JsonProperty("n_epochs")
     Integer nEpochs;
 
     /**

diff --git a/client/src/test/java/com/theokanning/openai/FineTuneTest.java b/client/src/test/java/com/theokanning/openai/FineTuneTest.java
@@ -38,6 +38,7 @@ void createFineTune() {
         FineTuneRequest request = FineTuneRequest.builder()
                 .trainingFile(fileId)
                 .model("ada")
+                .nEpochs(4)
                 .build();
 
         FineTuneResult fineTune = service.createFineTune(request);

diff --git a/service/src/test/java/com/theokanning/openai/service/FineTuneTest.java b/service/src/test/java/com/theokanning/openai/service/FineTuneTest.java
@@ -38,6 +38,7 @@ void createFineTune() {
         FineTuneRequest request = FineTuneRequest.builder()
                 .trainingFile(fileId)
                 .model("ada")
+                .nEpochs(4)
                 .build();
 
         FineTuneResult fineTune = service.createFineTune(request);
-Original file line number
+Diff line change
@@ Expand Up / @@ -37,6 +37,7 @@ public class FineTuneRequest { @@
         /**
          * The number of epochs to train the model for. An epoch refers to one full cycle through the training dataset.
          */
+        @JsonProperty("n_epochs")
         Integer nEpochs;
         /**
@@ Expand Down @@