RemoveClassNameMap in ReferenceField

marmelab · Aug 25, 2020 · d8e75ce · d8e75ce
1 parent 88ede7c
commit d8e75ce
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/packages/ra-ui-materialui/src/field/ReferenceField.tsx b/packages/ra-ui-materialui/src/field/ReferenceField.tsx
@@ -15,8 +15,8 @@ import {
 import LinearProgress from '../layout/LinearProgress';
 import Link from '../Link';
 import sanitizeRestProps from './sanitizeRestProps';
-import { ClassNameMap } from '@material-ui/styles';
 import { FieldProps, fieldPropTypes, InjectedFieldProps } from './types';
+import { ClassesOverride } from '../types';
 
 /**
  * Fetch reference record, and delegate rendering to child component.
@@ -131,7 +131,7 @@ ReferenceField.defaultProps = {
 
 export interface ReferenceFieldProps extends FieldProps, InjectedFieldProps {
     children: ReactElement;
-    classes?: Partial<ClassNameMap<ReferenceFieldClassKey>>;
+    classes?: ClassesOverride<typeof useStyles>;
     reference: string;
     resource?: string;
     source: string;
@@ -237,13 +237,11 @@ ReferenceFieldView.propTypes = {
     translateChoice: PropTypes.oneOfType([PropTypes.func, PropTypes.bool]),
 };
 
-export type ReferenceFieldClassKey = 'link';
-
 export interface ReferenceFieldViewProps
     extends FieldProps,
         InjectedFieldProps,
         UseReferenceProps {
-    classes?: Partial<ClassNameMap<ReferenceFieldClassKey>>;
+    classes?: ClassesOverride<typeof useStyles>;
     reference: string;
     resource?: string;
     translateChoice?: Function | boolean;