From c267cd60c156706991ec4c0e85c3167bddf8432a Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Tue, 20 Aug 2024 11:37:08 -0300
Subject: [PATCH] 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>
---
 .../GenericNode/components/NodeStatus/index.tsx            | 5 -----
 src/frontend/src/CustomNodes/GenericNode/index.tsx         | 7 ++++++-
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/frontend/src/CustomNodes/GenericNode/components/NodeStatus/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/NodeStatus/index.tsx
index 5951b7935..2ca06afcd 100644
--- 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
index bcbd2ce5e..25dc6c91f 100644
--- 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>