Apply suggestions from code review

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: nikagra

driver-core/src/main/java/com/datastax/driver/core/policies/TokenAwarePolicy.java

@@ -54,7 +54,7 @@
  *       HostDistance#LOCAL LOCAL} replicas for the query <em>if possible</em> (i.e. if the query's
  *       {@linkplain Statement#getRoutingKey(ProtocolVersion, CodecRegistry) routing key} is not
  *       {@code null} and if the {@linkplain Metadata#getReplicasList(String, String, Token.Factory,
- *       ByteBuffer)} (String, ByteBuffer) set of replicas} for that partition key is not empty). If
+ *       ByteBuffer) set of replicas} for that partition key is not empty). If
  *       no local replica can be either found or successfully contacted, the rest of the query plan
  *       will fall back to the child policy's one.
  * </ul>
@@ -81,9 +81,10 @@
  *       primary replica ordering from the token ring).
  *   <li>Remote replicas second: remaining replicas (typically in remote datacenters) are appended,
  *       but only if they are up and not ignored by the child policy.
- *   <li>No non-replica hosts: unlike regular queries, LWT plans do not fall back to non-replica
- *       hosts, ensuring all attempts target actual replicas to maintain LWT correctness and reduce
- *       coordinator forwarding overhead.
+ *   <li>Replica-only routing when possible: under normal conditions, LWT query plans target only
+ *       replicas for the partition in order to reduce coordinator forwarding overhead and improve
+ *       performance. When replica information is unavailable, the driver falls back to the child
+ *       policy as described in the fallback behavior below, which may include non-replica hosts.
  * </ul>
  *
  * <p><strong>Rack awareness</strong> is intentionally <em>not</em> applied to LWT replica ordering.

driver-core/src/test/java/com/datastax/driver/core/policies/TokenAwarePolicyTest.java

@@ -755,7 +755,7 @@ public void should_properly_generate_and_use_routing_key_for_composite_partition
         assertThat(rs.getExecutionInfo().getQueriedHost()).isEqualTo(host1);
         assertThat(rs.isExhausted()).isFalse();
         Row r = rs.one();
-        assertThat(rs.getExecutionInfo().getQueriedHost()).isEqualTo(host1);
+
         assertThat(rs.isExhausted()).isTrue();
 
         assertThat(r.getInt("i")).isEqualTo(3);