Allow sampling of fractional millis

2 files changed, 3 insertions, 4 deletions

diff --git a/linguistics/abi-spec.json b/linguistics/abi-spec.json
index 680aec3ff57..1ffb879e57e 100644
--- a/linguistics/abi-spec.json
+++ b/linguistics/abi-spec.json
@@ -373,7 +373,7 @@
       "abstract"
     ],
     "methods" : [
-      "public abstract void sampleEmbeddingLatency(java.time.Duration, com.yahoo.language.process.Embedder$Context)",
+      "public abstract void sampleEmbeddingLatency(double, com.yahoo.language.process.Embedder$Context)",
       "public abstract void sampleSequenceLength(long, com.yahoo.language.process.Embedder$Context)",
       "public static com.yahoo.language.process.Embedder$Runtime testInstance()"
     ],
diff --git a/linguistics/src/main/java/com/yahoo/language/process/Embedder.java b/linguistics/src/main/java/com/yahoo/language/process/Embedder.java
index 1d2e1bcc847..98030a4f054 100644
--- a/linguistics/src/main/java/com/yahoo/language/process/Embedder.java
+++ b/linguistics/src/main/java/com/yahoo/language/process/Embedder.java
@@ -6,7 +6,6 @@ import com.yahoo.language.Language;
 import com.yahoo.tensor.Tensor;
 import com.yahoo.tensor.TensorType;
 
-import java.time.Duration;
 import java.util.List;
 import java.util.Map;
 
@@ -72,13 +71,13 @@ public interface Embedder {
     @Beta
     interface Runtime {
         /** Sample latency metric for embedding */
-        void sampleEmbeddingLatency(Duration latency, Context ctx);
+        void sampleEmbeddingLatency(double millis, Context ctx);
         /** Sample sequence length metric for embedding */
         void sampleSequenceLength(long length, Context ctx);
 
         static Runtime testInstance() {
             return new Runtime() {
-                @Override public void sampleEmbeddingLatency(Duration latency, Context ctx) { }
+                @Override public void sampleEmbeddingLatency(double millis, Context ctx) { }
                 @Override public void sampleSequenceLength(long length, Context ctx) { }
             };
         }