From cdf69980f053c65837cb35a981277b86cfb0f801 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cle=CC=81ment=20Doumouro?= Date: Wed, 13 Nov 2024 13:47:22 +0100 Subject: [PATCH] fix: model size parameter name --- .../src/main/java/org/icij/datashare/nlp/NlpHelper.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/datashare-app/src/main/java/org/icij/datashare/nlp/NlpHelper.java b/datashare-app/src/main/java/org/icij/datashare/nlp/NlpHelper.java index 3486bae8f..bacf64976 100644 --- a/datashare-app/src/main/java/org/icij/datashare/nlp/NlpHelper.java +++ b/datashare-app/src/main/java/org/icij/datashare/nlp/NlpHelper.java @@ -6,7 +6,7 @@ public class NlpHelper { public static Map pipelineExtras(Pipeline.Type pipeline) { if (pipeline == Pipeline.Type.SPACY) { - return Map.of("size", "md"); + return Map.of("modelSize", "md"); } return Map.of(); }