handle coefficients as map (#57)

Author: aarsilv

pom.xml

@@ -6,7 +6,7 @@
 
     <groupId>cloud.eppo</groupId>
     <artifactId>eppo-server-sdk</artifactId>
-    <version>2.4.5</version>
+    <version>2.4.6</version>
 
     <name>${project.groupId}:${project.artifactId}</name>
     <description>Eppo Server-Side SDK for Java</description>

src/main/java/com/eppo/sdk/deserializer/BanditsDeserializer.java

@@ -53,9 +53,10 @@ public Map<String, BanditParameters> deserialize(
             JsonNode coefficientsNode = modelDataNode.get("coefficients");
 
             Map<String, BanditCoefficients> coefficients = new HashMap<>();
-            coefficientsNode.iterator().forEachRemaining(actionCoefficientsNode -> {
-                BanditCoefficients actionCoefficients = this.parseActionCoefficientsNode(actionCoefficientsNode);
-                coefficients.put(actionCoefficients.getActionKey(), actionCoefficients);
+            Iterator<Map.Entry<String, JsonNode>> coefficientIterator = coefficientsNode.fields();
+            coefficientIterator.forEachRemaining(field -> {
+                BanditCoefficients actionCoefficients = this.parseActionCoefficientsNode(field.getValue());
+                coefficients.put(field.getKey(), actionCoefficients);
             });
 
             modelData.setCoefficients(coefficients);

src/test/resources/bandits/bandits-parameters-1.json

@@ -10,8 +10,8 @@
         "gamma": 1.0,
         "defaultActionScore": 0.0,
         "actionProbabilityFloor": 0.0,
-        "coefficients": [
-          {
+        "coefficients": {
+          "nike": {
             "actionKey": "nike",
             "intercept": 1.0,
             "actionNumericCoefficients": [
@@ -50,7 +50,7 @@
               }
             ]
           },
-          {
+          "adidas": {
             "actionKey": "adidas",
             "intercept": 1.1,
             "actionNumericCoefficients": [
@@ -82,7 +82,7 @@
               }
             ]
           }
-        ]
+        }
       }
     },
     {
@@ -94,7 +94,7 @@
         "gamma": 1.0,
         "defaultActionScore": 0.0,
         "actionProbabilityFloor": 0.0,
-        "coefficients": []
+        "coefficients": {}
       }
     }
   ]