diff --git a/scheduler/data-flow/src/main/kotlin/io/seldon/dataflow/kafka/BatchProcessor.kt b/scheduler/data-flow/src/main/kotlin/io/seldon/dataflow/kafka/BatchProcessor.kt index 5596c3caa6..b3dad0b14c 100644 --- a/scheduler/data-flow/src/main/kotlin/io/seldon/dataflow/kafka/BatchProcessor.kt +++ b/scheduler/data-flow/src/main/kotlin/io/seldon/dataflow/kafka/BatchProcessor.kt @@ -1,6 +1,7 @@ package io.seldon.dataflow.kafka import io.klogging.noCoLogger +import io.seldon.mlops.inference.v2.V2Dataplane.InferParameter import io.seldon.mlops.inference.v2.V2Dataplane.InferTensorContents import io.seldon.mlops.inference.v2.V2Dataplane.ModelInferRequest import io.seldon.mlops.inference.v2.V2Dataplane.ModelInferRequest.InferInputTensor @@ -57,11 +58,13 @@ class BatchProcessor(private val threshold: Int) : Transformer() + v.map { it.parametersMap }.forEach {parameters.putAll(it)} InferInputTensor .newBuilder() .setName(k) .addAllShape(shape) + .putAllParameters(parameters) .setContents( InferTensorContents .newBuilder() @@ -84,6 +87,7 @@ class BatchProcessor(private val threshold: Int) : Transformer