diff --git a/li-apache-kafka-clients/src/main/java/com/linkedin/kafka/clients/producer/LiKafkaProducer.java b/li-apache-kafka-clients/src/main/java/com/linkedin/kafka/clients/producer/LiKafkaProducer.java
index db8342d..b681f6f 100644
--- a/li-apache-kafka-clients/src/main/java/com/linkedin/kafka/clients/producer/LiKafkaProducer.java
+++ b/li-apache-kafka-clients/src/main/java/com/linkedin/kafka/clients/producer/LiKafkaProducer.java
@@ -6,6 +6,10 @@
 
 import com.linkedin.kafka.clients.annotations.InterfaceOrigin;
 import java.util.concurrent.TimeUnit;
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+import org.apache.kafka.common.PartitionInfo;
 import org.apache.kafka.clients.producer.Producer;
 
 
@@ -23,4 +27,7 @@ public interface LiKafkaProducer<K, V> extends Producer<K, V> {
    */
   @InterfaceOrigin.LiKafkaClients
   void flush(long timeout, TimeUnit timeUnit);
+
+  @InterfaceOrigin.LiKafkaClients
+  Map<String, List<PartitionInfo>> partitionsFor(Set<String> topics);
 }
diff --git a/li-apache-kafka-clients/src/main/java/com/linkedin/kafka/clients/producer/LiKafkaProducerImpl.java b/li-apache-kafka-clients/src/main/java/com/linkedin/kafka/clients/producer/LiKafkaProducerImpl.java
index db87a25..1e5a523 100644
--- a/li-apache-kafka-clients/src/main/java/com/linkedin/kafka/clients/producer/LiKafkaProducerImpl.java
+++ b/li-apache-kafka-clients/src/main/java/com/linkedin/kafka/clients/producer/LiKafkaProducerImpl.java
@@ -351,6 +351,7 @@ public List<PartitionInfo> partitionsFor(String topic) {
     return _producer.partitionsFor(topic);
   }
 
+  @Override
   public Map<String, List<PartitionInfo>> partitionsFor(Set<String> topics) {
     //TODO come back here when upstream API settles
     throw new UnsupportedOperationException("not implemented");