From 182f6063f05d21d9ae27382eee645f45cda5c409 Mon Sep 17 00:00:00 2001
From: Piero Toffanin <pt@masseranolabs.com>
Date: Tue, 8 Apr 2025 22:09:08 -0400
Subject: [PATCH] ISO lang code support for --load-only

---
 libretranslate/init.py     | 2 ++
 libretranslate/language.py | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/libretranslate/init.py b/libretranslate/init.py
index 098cb20..0df8ba3 100644
--- a/libretranslate/init.py
+++ b/libretranslate/init.py
@@ -27,6 +27,8 @@ def check_and_install_models(force=False, load_only_lang_codes=None,update=False
         print("Found %s models" % len(available_packages))
         if load_only_lang_codes is not None:
             # load_only_lang_codes: List[str] (codes)
+            load_only_lang_codes = libretranslate.language.iso2model(load_only_lang_codes)
+
             # Ensure the user does not use any unavailable language code.
             unavailable_lang_codes = set(load_only_lang_codes)
             for pack in available_packages:
diff --git a/libretranslate/language.py b/libretranslate/language.py
index 7383c74..42f8866 100644
--- a/libretranslate/language.py
+++ b/libretranslate/language.py
@@ -14,6 +14,9 @@ aliases = {
 rev_aliases = {v.lower(): k for k, v in aliases.items()}
 
 def iso2model(lang):
+    if isinstance(lang, list):
+        return [iso2model(l) for l in lang]
+    
     if not isinstance(lang, str):
         return lang