Fix node size in nodeComponent (#3450)

* fix: set nide size directly in the nodeComponent * [autofix.ci] apply automated fixes --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
langflow-ai · Aug 20, 2024 · c267cd6 · c267cd6
1 parent ef08300
commit c267cd6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/frontend/src/CustomNodes/GenericNode/components/NodeStatus/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/NodeStatus/index.tsx
@@ -71,9 +71,6 @@ export default function NodeStatus({
     return frozen ? frozenClass : className;
   };
 
-  const getNodeSizeClass = (showNode) =>
-    showNode ? "w-96 rounded-lg" : "w-26 h-26 rounded-full";
-
   const getNodeBorderClassName = (
     selected: boolean,
     showNode: boolean,
@@ -87,10 +84,8 @@ export default function NodeStatus({
     );
 
     const baseBorderClass = getBaseBorderClass(selected);
-    const nodeSizeClass = getNodeSizeClass(showNode);
     const names = classNames(
       baseBorderClass,
-      nodeSizeClass,
       "generic-node-div group/node",
       specificClassFromBuildStatus,
     );

diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx
@@ -287,7 +287,12 @@ export default function GenericNode({
   return (
     <>
       {memoizedNodeToolbarComponent}
-      <div className={borderColor}>
+      <div
+        className={cn(
+          borderColor,
+          showNode ? "w-96 rounded-lg" : "w-26 h-26 rounded-full",
+        )}
+      >
         {data.node?.beta && showNode && (
           <div className="beta-badge-wrapper">
             <div className="beta-badge-content">BETA</div>