import hotfix (#1008)

gradio-app · Apr 15, 2022 · f511174 · f511174
1 parent 0e1dce6
commit f511174
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/gradio/components.py b/gradio/components.py
@@ -133,6 +133,9 @@ def get_component_shortcut(cls, str_shortcut: str) -> Optional[Component]:
             True, found_class or
             False, None
         """
+        # If we do not import templates Python cannot recognize grandchild classes names.
+        import gradio.templates
+
         # Make it suitable with class names
         str_shortcut = str_shortcut.replace("_", "")
         for sub_cls in cls.__subclasses__():

diff --git a/test/test_components.py b/test/test_components.py
@@ -22,6 +22,14 @@
 """
 
 
+class TestComponent(unittest.TestCase):
+    def test_component_functions(self):
+        """
+        component
+        """
+        assert isinstance(gr.component("text"), gr.templates.Text)
+
+
 class TestTextbox(unittest.TestCase):
     def test_component_functions(self):
         """