From 025873df2c48933a6d31f4fa6f49e185bcaa0029 Mon Sep 17 00:00:00 2001
From: Kristoffer Andersson <kristoffer.andersson@gu.se>
Date: Tue, 26 Mar 2024 15:00:45 +0100
Subject: [PATCH] fix: add language to annotation

Fixes #32
---
 .../src/sbx_ocr_correction_viklofg_sweocr/__init__.py         | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/ocr-correction-viklofg-sweocr/src/sbx_ocr_correction_viklofg_sweocr/__init__.py b/ocr-correction-viklofg-sweocr/src/sbx_ocr_correction_viklofg_sweocr/__init__.py
index ea50b03..9cfb75a 100644
--- a/ocr-correction-viklofg-sweocr/src/sbx_ocr_correction_viklofg_sweocr/__init__.py
+++ b/ocr-correction-viklofg-sweocr/src/sbx_ocr_correction_viklofg_sweocr/__init__.py
@@ -27,9 +27,7 @@
 TOK_SEP = " "
 
 
-@annotator(
-    "Word neighbour tagging with a masked Bert model",
-)
+@annotator("Word neighbour tagging with a masked Bert model", language=["swe"])
 def annotate_ocr_correction(
     out_ocr_correction: Output = Output(
         "<token>:sbx_ocr_correction_viklofg_sweocr.ocr-correction--viklofg-sweocr",