[yt provider] Fix YtCopy type annotation check for column groups (#11602

)
ydb-platform · Nov 14, 2024 · 85bb210 · 85bb210
1 parent f00baa4
commit 85bb210
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ydb/library/yql/providers/yt/provider/yql_yt_datasink_type_ann.cpp b/ydb/library/yql/providers/yt/provider/yql_yt_datasink_type_ann.cpp
@@ -965,6 +965,10 @@ class TYtDataSinkTypeAnnotationTransformer : public TVisitorTransformerBase {
             if (auto setting = NYql::GetSetting(GetOutputOp(out.Cast()).Output().Item(FromString<ui32>(out.Cast().OutIndex().Value())).Settings().Ref(), EYtSettingType::ColumnGroups)) {
                 inputColGroupSpec = setting->Tail().Content();
             }
+        } else if (auto outTable = path.Table().Maybe<TYtOutTable>()) {
+            if (auto setting = NYql::GetSetting(outTable.Cast().Settings().Ref(), EYtSettingType::ColumnGroups)) {
+                inputColGroupSpec = setting->Tail().Content();
+            }
         }
 
         if (outGroup != inputColGroupSpec) {