fix: set data methods for queries (typings)

Author: js2me

src/mobx-inifinite-query.ts

@@ -12,6 +12,7 @@ import {
   InfiniteData,
   RefetchOptions,
   SetDataOptions,
+  Updater,
 } from '@tanstack/query-core';
 import { IDisposer } from 'disposer-util';
 import { LinkedAbortController } from 'linked-abort-controller';
@@ -249,14 +250,15 @@ export class MobxInfiniteQuery<
   }
 
   setData(
-    data:
-      | InfiniteData<TData>
-      | ((input: InfiniteData<TData> | undefined) => InfiniteData<TData>),
+    updater: Updater<
+      NoInfer<InfiniteData<TData>> | undefined,
+      NoInfer<InfiniteData<TData>> | undefined
+    >,
     options?: SetDataOptions,
   ) {
     this.queryClient.setQueryData<InfiniteData<TData>>(
       this.options.queryKey,
-      data,
+      updater,
       options,
     );
   }

src/mobx-query.ts

@@ -11,6 +11,7 @@ import {
   QueryObserverResult,
   RefetchOptions,
   SetDataOptions,
+  Updater,
 } from '@tanstack/query-core';
 import { IDisposer } from 'disposer-util';
 import { LinkedAbortController } from 'linked-abort-controller';
@@ -212,10 +213,14 @@ export class MobxQuery<
   }
 
   setData(
-    data: TData | ((input: TData | undefined) => TData),
+    updater: Updater<NoInfer<TData> | undefined, NoInfer<TData> | undefined>,
     options?: SetDataOptions,
   ) {
-    this.queryClient.setQueryData<TData>(this.options.queryKey, data, options);
+    this.queryClient.setQueryData<TData>(
+      this.options.queryKey,
+      updater,
+      options,
+    );
   }
 
   update(options: Partial<QueryObserverOptions<TData, TError, TQueryKey>>) {