[DataGridPro] Fix the return type of useGridApiContext() for Pro an…

…d Premium packages on React < 19 (#16441)
mui · Feb 4, 2025 · e9be335 · e9be335
1 parent 1e6ac1b
commit e9be335
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 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
@@ -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
@@ -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;