Merge pull request #22 from mirecl/21-migrate-to-catboost-126

Migrate to catboost v1.2.6
mirecl · Sep 5, 2024 · cd0b8a9 · cd0b8a9
2 parents f80b7c8 + 0999cae
commit cd0b8a9
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 7 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -5,6 +5,9 @@ on:
   pull_request:
     branches: ["master"]
 
+env:
+  CATBOOST_VERSION: "v1.2.6"
+
 jobs:
   Linter:
     runs-on: ubuntu-latest
@@ -51,7 +54,7 @@ jobs:
 
       - name: Download library
         run: |
-          sudo wget -q "https://github.com/catboost/catboost/releases/download/v1.2.5/libcatboostmodel.so" -O /usr/local/lib/libcatboostmodel.so
+          sudo wget -q "https://github.com/catboost/catboost/releases/download/$CATBOOST_VERSION/libcatboostmodel.so" -O /usr/local/lib/libcatboostmodel.so
 
       - name: Run tests
         run: |
@@ -82,12 +85,12 @@ jobs:
         if: matrix.platform == 'macos-latest'
         run: |
           sudo mkdir /usr/local/lib
-          sudo wget -q "https://github.com/catboost/catboost/releases/download/v1.2.5/libcatboostmodel.dylib" -O /usr/local/lib/libcatboostmodel.dylib
+          sudo wget -q "https://github.com/catboost/catboost/releases/download/$CATBOOST_VERSION/libcatboostmodel.dylib" -O /usr/local/lib/libcatboostmodel.dylib
 
       - name: Configuration Linux
         if: matrix.platform == 'ubuntu-latest'
         run: |
-          sudo wget -q "https://github.com/catboost/catboost/releases/download/v1.2.5/libcatboostmodel.so" -O /usr/local/lib/libcatboostmodel.so
+          sudo wget -q "https://github.com/catboost/catboost/releases/download/$CATBOOST_VERSION/libcatboostmodel.so" -O /usr/local/lib/libcatboostmodel.so
 
       - name: Set up Go
         uses: actions/setup-go@v5

diff --git a/README.md b/README.md
@@ -25,13 +25,15 @@ For more information, see <https://catboost.ai/en/docs/concepts/c-plus-plus-api_
     <th>v1.2.3</th>
     <th>v1.2.4</th>
     <th>v1.2.5</th>
+    <th>v1.2.6</th>
   </tr>
   <tr>
     <td align="center">🚫 (not testing)</td>
     <td align="center">✅</td>
     <td align="center">✅</td>
     <td align="center">✅</td>
     <td align="center">✅</td>
+    <td align="center">✅</td>
   </tr>
 </table>
 

diff --git a/catboost/c_api.h b/catboost/c_api.h
@@ -6,7 +6,7 @@
 
 #define CATBOOST_APPLIER_MAJOR 1
 #define CATBOOST_APPLIER_MINOR 2
-#define CATBOOST_APPLIER_FIX 5
+#define CATBOOST_APPLIER_FIX 6
 
 #if defined(__cplusplus)
 extern "C" {
@@ -53,6 +53,7 @@ enum EApiPredictionType {
     APT_RMSE_WITH_UNCERTAINTY = 2,
     APT_PROBABILITY = 3,
     APT_CLASS = 4,
+    APT_MULTI_PROBABILITY = 5,
 };
 
 enum ECatBoostApiFormulaEvaluatorType {
@@ -102,7 +103,7 @@ CATBOOST_API bool LoadFullModelFromBuffer(
     size_t binaryBufferSize);
 
 /**
- * Use CUDA gpu device for model evaluation
+ * Use CUDA GPU device for model evaluation
 */
 CATBOOST_API bool EnableGPUEvaluation(ModelCalcerHandle* modelHandle, int deviceId);
 
@@ -672,4 +673,4 @@ CATBOOST_API bool GetModelUsedFeaturesNames(ModelCalcerHandle* modelHandle, char
 
 #if defined(__cplusplus)
 }
-#endif
+#endif
diff --git a/catboost/catboost_test.go b/catboost/catboost_test.go
@@ -16,7 +16,7 @@ const (
 )
 
 func TestVersion(t *testing.T) {
-	require.Equal(t, "v1.2.5", cb.Version())
+	require.Equal(t, "v1.2.6", cb.Version())
 }
 
 func TestLoadFullModel(t *testing.T) {