Types fix

Author: VerZsuT

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "react-afc",
-  "version": "3.4.0",
+  "version": "3.4.1",
   "description": "Allows you to significantly simplify the optimization of functional react components",
   "keywords": [
     "react",

package.json

@@ -9,7 +9,7 @@ import type { Actions, ObjectState, ObjectStateSetters, ReduxSelectors, State }
 import * as AFC from './index'
 
 /** Allow to use inline callbacks */
-export function $<T extends (...args: any[]) => any>(callback: T) {
+export function $<T extends (...args: any[]) => any>(callback: T): T {
   if (inAFCRender()) return AFC.$(callback)
   return callback
 }

src/compatible.ts

@@ -66,7 +66,7 @@ export function afcMemo<P extends object>(constructor: Constructor<P>, options?:
 }
 
 /** Allow to use inline callbacks */
-export function $<T extends (...args: any[]) => any>(callback: T) {
+export function $<T extends (...args: any[]) => any>(callback: T): T {
   const { callbacks } = getRenderData()
 
   if (callbacks.arr.length <= callbacks.next) {
@@ -75,7 +75,7 @@ export function $<T extends (...args: any[]) => any>(callback: T) {
     return callback
   }
   else {
-    return callbacks.arr[callbacks.next++]
+    return callbacks.arr[callbacks.next++] as T
   }
 }