bug when torch not installed

kornia · Mar 12, 2023 · 911a158 · 911a158
1 parent 5a37f08
commit 911a158
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/limbus/core/component.py b/limbus/core/component.py
@@ -22,15 +22,15 @@
 log = logging.getLogger(__name__)
 
 
+base_class: Type = object
 if config.COMPONENT_TYPE == "generic":
-    base_class: Type = object
+    pass
 elif config.COMPONENT_TYPE == "torch":
     try:
         base_class = nn.Module
     except NameError:
         log.error("Torch not installed. Using generic base class.")
 else:
-    base_class = object
     log.error("Invalid component type. Using generic base class.")