Use package.version for model version comparison

Change `--install-models` to `--force-update-models`
This commit is contained in:
Robert George 2023-09-29 15:31:33 -07:00
parent 78713d8017
commit 94f63cedda
No known key found for this signature in database
GPG key ID: 9B0D835A293663C4
5 changed files with 6 additions and 4 deletions

View file

@ -1,5 +1,6 @@
from argostranslate import package, translate
from packaging import version
import libretranslate.language
@ -55,7 +56,7 @@ def check_and_install_models(force=False, load_only_lang_codes=None,update=False
and pack.to_code == available_package.to_code
):
update = True
if pack.package_version < available_package.package_version:
if version.parse(pack.package_version) < version.parse(available_package.package_version):
print(
f"Updating {available_package} ({pack.package_version}->{available_package.package_version}) ..."
)