From e9be335d4db5089b4db5b5efb0f4fc8263fe7550 Mon Sep 17 00:00:00 2001
From: Armin Mehinovic <4390250+arminmeh@users.noreply.github.com>
Date: Tue, 4 Feb 2025 08:52:45 +0100
Subject: [PATCH] [DataGridPro] Fix the return type of `useGridApiContext()`
 for Pro and Premium packages on React < 19 (#16441)

---
 .../src/hooks/utils/useGridApiContext.ts                    | 6 ++++--
 .../x-data-grid-pro/src/hooks/utils/useGridApiContext.ts    | 6 ++++--
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/packages/x-data-grid-premium/src/hooks/utils/useGridApiContext.ts b/packages/x-data-grid-premium/src/hooks/utils/useGridApiContext.ts
index de89813383fde..a998dced7aa03 100644
--- a/packages/x-data-grid-premium/src/hooks/utils/useGridApiContext.ts
+++ b/packages/x-data-grid-premium/src/hooks/utils/useGridApiContext.ts
@@ -1,4 +1,6 @@
-import { useGridApiContext as useCommunityGridApiContext } from '@mui/x-data-grid';
+import { RefObject } from '@mui/x-internals/types';
+import { GridApiCommon, useGridApiContext as useCommunityGridApiContext } from '@mui/x-data-grid';
 import { GridApiPremium } from '../../models/gridApiPremium';
 
-export const useGridApiContext = useCommunityGridApiContext<GridApiPremium>;
+export const useGridApiContext: <Api extends GridApiCommon = GridApiPremium>() => RefObject<Api> =
+  useCommunityGridApiContext;
diff --git a/packages/x-data-grid-pro/src/hooks/utils/useGridApiContext.ts b/packages/x-data-grid-pro/src/hooks/utils/useGridApiContext.ts
index bd5f4f2a1b599..d3ee820aac6d4 100644
--- a/packages/x-data-grid-pro/src/hooks/utils/useGridApiContext.ts
+++ b/packages/x-data-grid-pro/src/hooks/utils/useGridApiContext.ts
@@ -1,4 +1,6 @@
-import { useGridApiContext as useCommunityGridApiContext } from '@mui/x-data-grid';
+import { RefObject } from '@mui/x-internals/types';
+import { GridApiCommon, useGridApiContext as useCommunityGridApiContext } from '@mui/x-data-grid';
 import { GridApiPro } from '../../models/gridApiPro';
 
-export const useGridApiContext = useCommunityGridApiContext<GridApiPro>;
+export const useGridApiContext: <Api extends GridApiCommon = GridApiPro>() => RefObject<Api> =
+  useCommunityGridApiContext;