apply css and update stories

Author: mykalmax

web/common/src/components/Lineage/LineageLayout.tsx

@@ -66,13 +66,15 @@ export function LineageLayout<
 }) {
   return (
     <ReactFlowProvider>
-      <LineageLayoutContainer isBuildingLayout={isBuildingLayout}>
+      <LineageLayoutContainer
+        isBuildingLayout={isBuildingLayout}
+        className={className}
+      >
         <LineageLayoutBase
           nodeTypes={nodeTypes}
           edgeTypes={edgeTypes}
           nodesDraggable={nodesDraggable}
           nodesConnectable={nodesConnectable}
-          className={className}
           controls={controls}
           useLineage={useLineage}
           onNodeClick={onNodeClick}

web/common/src/components/Lineage/LineageLayoutContainer.tsx

@@ -19,7 +19,7 @@ export function LineageLayoutContainer({
 }) {
   return (
     <VerticalContainer
-      data-component="LineageLayout"
+      data-component="LineageLayoutContainer"
       className={cn(
         'border-2 border-lineage-border bg-lineage-background relative h-full',
         className,

web/common/src/components/Lineage/stories/ModelLineage.tsx

@@ -383,6 +383,7 @@ export const ModelLineage = ({
         ModelEdgeId,
         ModelColumnID
       >
+        isBuildingLayout={isBuildingLayout}
         useLineage={useModelLineage}
         nodeTypes={nodeTypes}
         edgeTypes={edgeTypes}