From 91cbe5bc0133cf01abf1a39a43af7f7fd8457d55 Mon Sep 17 00:00:00 2001
From: Sidharth Vinod <sidharthv96@gmail.com>
Date: Tue, 18 Feb 2025 00:29:57 +0530
Subject: [PATCH] fix: StateStmt import

---
 packages/mermaid/src/diagrams/state/dataFetcher.ts | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/packages/mermaid/src/diagrams/state/dataFetcher.ts b/packages/mermaid/src/diagrams/state/dataFetcher.ts
index b390b9d65..d15217ae3 100644
--- a/packages/mermaid/src/diagrams/state/dataFetcher.ts
+++ b/packages/mermaid/src/diagrams/state/dataFetcher.ts
@@ -34,7 +34,7 @@ import {
   STMT_RELATION,
   STMT_STATE,
 } from './stateCommon.js';
-import type { Edge, NodeData, State, StateStmt, Stmt, StyleClass } from './stateDb.js';
+import type { Edge, NodeData, StateStmt, Stmt, StyleClass } from './stateDb.js';
 
 // List of nodes created from the parsed diagram statement items
 const nodeDb = new Map<string, NodeData>();
@@ -59,7 +59,7 @@ export function stateDomId(
 const setupDoc = (
   parentParsedItem: StateStmt | undefined,
   doc: Stmt[],
-  diagramStates: Map<string, State>,
+  diagramStates: Map<string, StateStmt>,
   nodes: NodeData[],
   edges: Edge[],
   altFlag: boolean,
@@ -177,18 +177,18 @@ function insertOrUpdateNode(
  * Else create 1 string from the list of classes found
  *
  */
-function getClassesFromDbInfo(dbInfoItem?: State): string {
+function getClassesFromDbInfo(dbInfoItem?: StateStmt): string {
   return dbInfoItem?.classes?.join(' ') ?? '';
 }
 
-function getStylesFromDbInfo(dbInfoItem?: State): string[] {
+function getStylesFromDbInfo(dbInfoItem?: StateStmt): string[] {
   return dbInfoItem?.styles ?? [];
 }
 
 export const dataFetcher = (
   parent: StateStmt | undefined,
   parsedItem: StateStmt,
-  diagramStates: Map<string, State>,
+  diagramStates: Map<string, StateStmt>,
   nodes: NodeData[],
   edges: Edge[],
   altFlag: boolean,