diff --git a/src/createMemo.ts b/src/createMemo.ts index 1a95dd06..9b796877 100644 --- a/src/createMemo.ts +++ b/src/createMemo.ts @@ -1,5 +1,6 @@ import { useMemo } from 'react'; -const createMemo = fn => (...args) => useMemo(() => fn(...args), args); +const createMemo = any>(fn: T) => (...args: Parameters) => + useMemo>(() => fn(...args), args); export default createMemo;