fix: remove no-else-return block in _get_legend_line

Author: jkasiraj

src/sagemaker/lineage/query.py

@@ -324,14 +324,14 @@ def _get_legend_line(self, component_name):
                 color=self.graph_styles[component_name]["style"]["background-color"],
                 name=self.graph_styles[component_name]["name"],
             )
-        else:
-            return '<div style="background-color: #ffffff; width: 1.6vw; height: 1.6vw;\
-                display: inline-block; font-size: 0.9vw; vertical-align: -0.2em;">{shape}</div>\
-                <div style="width: 0.3vw; height: 1.5vw; display: inline-block;"></div>\
-                <div style="display: inline-block; font-size: 1.5vw;">{name}</div></div>'.format(
-                shape=self.graph_styles[component_name]["style"]["shape"],
-                name=self.graph_styles[component_name]["name"],
-            )
+
+        return '<div style="background-color: #ffffff; width: 1.6vw; height: 1.6vw;\
+            display: inline-block; font-size: 0.9vw; vertical-align: -0.2em;">{shape}</div>\
+            <div style="width: 0.3vw; height: 1.5vw; display: inline-block;"></div>\
+            <div style="display: inline-block; font-size: 1.5vw;">{name}</div></div>'.format(
+            shape=self.graph_styles[component_name]["style"]["shape"],
+            name=self.graph_styles[component_name]["name"],
+        )
 
     def _add_legend(self, path):
         """Embed legend to html file generated by pyvis."""