From a735113466a0b56d028dd1de9dfa585b1f0d3c73 Mon Sep 17 00:00:00 2001
From: LordBaryhobal <lordbaryhobal@gmail.com>
Date: Mon, 25 May 2026 12:46:04 +0200
Subject: [PATCH] fix(parser): update ast gen script

---
 gen/gen.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/gen/gen.py b/gen/gen.py
index 106dcc7..34781b3 100644
--- a/gen/gen.py
+++ b/gen/gen.py
@@ -14,7 +14,8 @@ from abc import ABC, abstractmethod
 from dataclasses import dataclass
 from typing import Any, Generic, Optional, TypeVar
 
-from lexer.token import Token
+from midas.ast.location import Location
+from midas.lexer.token import Token
 
 T = TypeVar("T")
 
@@ -23,8 +24,10 @@ T = TypeVar("T")
 ##############
 
 
-@dataclass(frozen=True)
+@dataclass(frozen=True, kw_only=True)
 class Stmt(ABC):
+    location: Optional[Location] = None
+
     @abstractmethod
     def accept(self, visitor: Visitor[T]) -> T: ...
 
@@ -40,8 +43,10 @@ class Stmt(ABC):
 ###############
 
 
-@dataclass(frozen=True)
+@dataclass(frozen=True, kw_only=True)
 class Expr(ABC):
+    location: Optional[Location] = None
+
     @abstractmethod
     def accept(self, visitor: Visitor[T]) -> T: ...