bugfix in setting license; add 6mm to download of total model

wasserth · Jan 19, 2024 · 9c35040 · 9c35040
1 parent 5162531
commit 9c35040
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/totalsegmentator/bin/totalseg_download_weights.py b/totalsegmentator/bin/totalseg_download_weights.py
@@ -26,7 +26,7 @@ def main():
     args = parser.parse_args()
 
     task_to_id = {
-        "total": [291, 292, 293, 294, 295],
+        "total": [291, 292, 293, 294, 295, 298],
         "total_fast": [297, 298],
         "lung_vessels": [258],
         "cerebral_bleed": [150],

diff --git a/totalsegmentator/bin/totalseg_set_license.py b/totalsegmentator/bin/totalseg_set_license.py
@@ -21,8 +21,8 @@ def main():
 
     args = parser.parse_args()
 
-    if not args.license_number.startswith("aca_") or not args.license_number.startswith("com_"):
-        raise ValueError("license number must start with 'aca_' or 'com_' ")
+    if not args.license_number.startswith("aca_"):
+        raise ValueError("license number must start with 'aca_'")
     if len(args.license_number) != 18:
         raise ValueError("license number must have exactly 18 characters.")