fix edge case that breaches both max and decimal places

d2/src/main/java/com/linkedin/d2/balancer/servers/ZooKeeperAnnouncer.java

@@ -246,6 +246,8 @@ public ZooKeeperAnnouncer(LoadBalancerServer server, boolean initialIsUp,
     _warmupDuration = warmupDuration;
     _executorService = executorService;
     _eventEmitter = eventEmitter;
+
+    // take in max weight as a string to be precise at decimal places when creating a BigDecimal.
     _maxWeight = maxWeight == null ? null : new BigDecimal(maxWeight);
     if (actOnWeightBreach != null)
     {
@@ -873,6 +875,7 @@ private Map<Integer, PartitionData> validatePartitionData(Map<Integer, Partition
             throw getMaxWeightBreachException(weight, entry.getKey());
           case RECTIFY:
             entry.setValue(new PartitionData(_maxWeight.intValue()));
+            weight = _maxWeight;
             _log.warn("Capped weight {} in Partition {} to the max weight allowed: {}.", weight.doubleValue(),
                 entry.getKey(), _maxWeight.intValue());
             break;