From 911a15817bb0527b0504e711269fa7f9d0df8e80 Mon Sep 17 00:00:00 2001
From: Luis Ferraz <luisferrazc@gmail.com>
Date: Mon, 13 Mar 2023 00:53:26 +0100
Subject: [PATCH] bug when torch not installed

---
 limbus/core/component.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/limbus/core/component.py b/limbus/core/component.py
index fa3c9a4..d01d381 100644
--- 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.")