[Badge][material-next] Apply new OwnerState type to Badge (#40119)

mui · Dec 6, 2023 · d56d14e · d56d14e
1 parent dc77e53
commit d56d14e
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/packages/mui-material-next/src/Badge/Badge.types.ts b/packages/mui-material-next/src/Badge/Badge.types.ts
@@ -1,6 +1,6 @@
 import * as React from 'react';
 import { SxProps } from '@mui/system';
-import { OverridableStringUnion, OverrideProps } from '@mui/types';
+import { OverridableStringUnion, OverrideProps, PartiallyRequired } from '@mui/types';
 import { SlotComponentProps } from '@mui/base';
 import { Theme } from '../styles';
 import { BadgeClasses } from './badgeClasses';
@@ -126,10 +126,8 @@ export type BadgeProps<
   AdditionalProps = {},
 > = OverrideProps<BadgeTypeMap<RootComponent, AdditionalProps>, RootComponent>;
 
-export interface BadgeOwnerState extends BadgeProps {
-  size: NonNullable<BadgeProps['size']>;
-  variant: NonNullable<BadgeProps['variant']>;
-  anchorOrigin: NonNullable<BadgeProps['anchorOrigin']>;
-  overlap: NonNullable<BadgeProps['overlap']>;
-  color: NonNullable<BadgeProps['color']>;
-}
+export interface BadgeOwnerState
+  extends PartiallyRequired<
+    BadgeProps,
+    'anchorOrigin' | 'color' | 'overlap' | 'size' | 'variant'
+  > {}