diff options
Diffstat (limited to 'linguistics')
-rw-r--r-- | linguistics/abi-spec.json | 2 | ||||
-rw-r--r-- | linguistics/src/main/java/com/yahoo/language/process/Embedder.java | 5 |
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) { } }; } |