From f42f9581736dc45fa34637363bb953e38fe84237 Mon Sep 17 00:00:00 2001
From: Vitaly Stoyan <vvvv@ydb.tech>
Date: Tue, 22 Oct 2024 20:36:59 +0300
Subject: [PATCH] build

---
 ydb/library/yql/core/ut/yql_expr_constraint_ut.cpp | 3 +--
 ydb/library/yql/core/ut/yql_expr_discover_ut.cpp   | 3 +--
 ydb/library/yql/core/ut/yql_expr_optimize_ut.cpp   | 3 +--
 ydb/library/yql/core/ut/yql_expr_providers_ut.cpp  | 3 +--
 4 files changed, 4 insertions(+), 8 deletions(-)

diff --git a/ydb/library/yql/core/ut/yql_expr_constraint_ut.cpp b/ydb/library/yql/core/ut/yql_expr_constraint_ut.cpp
index df3fdef8e5b3..cada31f156d2 100644
--- a/ydb/library/yql/core/ut/yql_expr_constraint_ut.cpp
+++ b/ydb/library/yql/core/ut/yql_expr_constraint_ut.cpp
@@ -31,9 +31,8 @@ Y_UNIT_TEST_SUITE(TYqlExprConstraints) {
         auto ytGateway = CreateYtFileGateway(yqlNativeServices);
         auto typeAnnotationContext = MakeIntrusive<TTypeAnnotationContext>();
         typeAnnotationContext->RandomProvider = CreateDeterministicRandomProvider(1);
-        auto ytState = MakeIntrusive<TYtState>();
+        auto ytState = MakeIntrusive<TYtState>(typeAnnotationContext.Get());
         ytState->Gateway = ytGateway;
-        ytState->Types = typeAnnotationContext.Get();
 
         InitializeYtGateway(ytGateway, ytState);
         typeAnnotationContext->AddDataSink(YtProviderName, CreateYtDataSink(ytState));
diff --git a/ydb/library/yql/core/ut/yql_expr_discover_ut.cpp b/ydb/library/yql/core/ut/yql_expr_discover_ut.cpp
index b754ee74433a..db6bf289f3ec 100644
--- a/ydb/library/yql/core/ut/yql_expr_discover_ut.cpp
+++ b/ydb/library/yql/core/ut/yql_expr_discover_ut.cpp
@@ -31,9 +31,8 @@ Y_UNIT_TEST_SUITE(TDiscoverYqlExpr) {
         auto ytGateway = CreateYtFileGateway(yqlNativeServices);
         auto typeAnnotationContext = MakeIntrusive<TTypeAnnotationContext>();
         typeAnnotationContext->DiscoveryMode = true;
-        auto ytState = MakeIntrusive<TYtState>();
+        auto ytState = MakeIntrusive<TYtState>(typeAnnotationContext.Get());
         ytState->Gateway = ytGateway;
-        ytState->Types = typeAnnotationContext.Get();
 
         InitializeYtGateway(ytGateway, ytState);
         auto randomProvider = CreateDeterministicRandomProvider(1);
diff --git a/ydb/library/yql/core/ut/yql_expr_optimize_ut.cpp b/ydb/library/yql/core/ut/yql_expr_optimize_ut.cpp
index 024e0b437b1c..bae3f0250c76 100644
--- a/ydb/library/yql/core/ut/yql_expr_optimize_ut.cpp
+++ b/ydb/library/yql/core/ut/yql_expr_optimize_ut.cpp
@@ -592,9 +592,8 @@ Y_UNIT_TEST_SUITE(TOptimizeYqlExpr) {
         auto yqlNativeServices = NFile::TYtFileServices::Make(functionRegistry.Get(), testTables);
         auto ytGateway = CreateYtFileGateway(yqlNativeServices);
         auto typeAnnotationContext = MakeIntrusive<TTypeAnnotationContext>();
-        auto ytState = MakeIntrusive<TYtState>();
+        auto ytState = MakeIntrusive<TYtState>(typeAnnotationContext.Get());
         ytState->Gateway = ytGateway;
-        ytState->Types = typeAnnotationContext.Get();
 
         InitializeYtGateway(ytGateway, ytState);
         auto randomProvider = CreateDeterministicRandomProvider(1);
diff --git a/ydb/library/yql/core/ut/yql_expr_providers_ut.cpp b/ydb/library/yql/core/ut/yql_expr_providers_ut.cpp
index 9cc6153e860d..ed99de46abe2 100644
--- a/ydb/library/yql/core/ut/yql_expr_providers_ut.cpp
+++ b/ydb/library/yql/core/ut/yql_expr_providers_ut.cpp
@@ -59,9 +59,8 @@ Y_UNIT_TEST_SUITE(TCompileWithProvidersYqlExpr) {
         auto yqlNativeServices = NFile::TYtFileServices::Make(functionRegistry.Get(), testTables);
         auto ytGateway = CreateYtFileGateway(yqlNativeServices);
         auto typeAnnotationContext = MakeIntrusive<TTypeAnnotationContext>();
-        auto ytState = MakeIntrusive<TYtState>();
+        auto ytState = MakeIntrusive<TYtState>(typeAnnotationContext.Get());
         ytState->Gateway = ytGateway;
-        ytState->Types = typeAnnotationContext.Get();
 
         InitializeYtGateway(ytGateway, ytState);
         typeAnnotationContext->AddDataSink(YtProviderName, CreateYtDataSink(ytState));