Remove single comma that prevented Stanza from using GPU 😩

spraakbanken · Oct 27, 2023 · 256b1bf · 256b1bf
1 parent 3f241a0
commit 256b1bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sparv/modules/stanza/stanza_swe.py b/sparv/modules/stanza/stanza_swe.py
@@ -133,7 +133,7 @@ def annotate_swe(
             logger.debug(f"Running dependency parsing and POS-taggning on {len(sentences)} sentences"
                          f" (using {'GPU' if use_gpu and not fallback else 'CPU'}).")
             nlp_args["processors"] = "tokenize,pos,lemma,depparse"  # Comma-separated list of processors to use
-            nlp_args["use_gpu"] = use_gpu and not fallback,
+            nlp_args["use_gpu"] = use_gpu and not fallback
             nlp = stanza.Pipeline(**nlp_args)
 
         else: