From d9dbd360ed24741d18fa413a4403e6ef618026ef Mon Sep 17 00:00:00 2001
From: NiharDoshi99 <51595473+NiharDoshi99@users.noreply.github.com>
Date: Fri, 10 Feb 2023 19:55:50 +0530
Subject: [PATCH] fix: added tag in datamodel generation (#10179)

* fix: added tag in datamodel generation

* fix: removed location
---
 scripts/datamodel_generation.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/scripts/datamodel_generation.py b/scripts/datamodel_generation.py
index bbacb86ec03..a58ddc796dc 100644
--- a/scripts/datamodel_generation.py
+++ b/scripts/datamodel_generation.py
@@ -19,8 +19,7 @@ import datamodel_code_generator.model.pydantic
 from datamodel_code_generator.imports import Import
 
 UNICODE_REGEX_REPLACEMENT_FILE_PATHS = [
-    "ingestion/src/metadata/generated/schema/entity/data/location.py",
-    "ingestion/src/metadata/generated/schema/entity/data/table.py",
+    "ingestion/src/metadata/generated/schema/entity/classification/tag.py",
     "ingestion/src/metadata/generated/schema/entity/events/webhook.py",
     "ingestion/src/metadata/generated/schema/entity/teams/user.py",
     "ingestion/src/metadata/generated/schema/entity/type.py",