Update vision_transformer_sam.py

Smartappli · May 25, 2024 · 732afcf · 732afcf
1 parent f829acf
commit 732afcf
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/fl_common/models/vision_transformer_sam.py b/fl_common/models/vision_transformer_sam.py
@@ -23,9 +23,8 @@ def get_vision_transformer_sam_model(vision_transformer_sam_type, num_classes):
     }
 
     if vision_transformer_sam_type not in valid_types:
-        raise ValueError(
-            f"Unknown Vision Transformer SAM Architecture: {vision_transformer_sam_type}"
-        )
+        msg = f"Unknown Vision Transformer SAM Architecture: {vision_transformer_sam_type}"
+        raise ValueError(msg)
 
     try:
         return create_model(