fix(types): use new createWithEqualityFn signature

packages/fiber/src/core/hooks.tsx

@@ -35,8 +35,7 @@ export function useThree<T = RootState>(
   selector: (state: RootState) => T = (state) => state as unknown as T,
   equalityFn?: <T>(state: T, newState: T) => boolean,
 ): T {
-  // TODO: fix this type
-  return useStore()(selector, equalityFn as any)
+  return useStore()(selector, equalityFn)
 }
 
 /**

packages/fiber/src/core/renderer.tsx

@@ -1,7 +1,7 @@
 import * as THREE from 'three'
 import * as React from 'react'
 import { ConcurrentRoot } from 'react-reconciler/constants'
-import { create } from 'zustand'
+import { createWithEqualityFn } from 'zustand/traditional'
 
 import type { Properties, ThreeElement } from '../three-types'
 import {
@@ -569,7 +569,7 @@ function Portal({ state = {}, children, container }: PortalProps): JSX.Element {
   })
 
   const usePortalStore = React.useMemo(() => {
-    const store = create<RootState>((set, get) => ({ ...rest, set, get } as RootState))
+    const store = createWithEqualityFn<RootState>((set, get) => ({ ...rest, set, get } as RootState))
 
     // Subscribe to previous root-state and copy changes over to the mirrored portal-state
     const onMutate = (prev: RootState) => store.setState((state) => inject.current(prev, state))

packages/fiber/src/core/store.ts

@@ -1,6 +1,7 @@
 import * as THREE from 'three'
 import * as React from 'react'
-import { create, type StoreApi, type UseBoundStore } from 'zustand'
+import { type StoreApi } from 'zustand'
+import { createWithEqualityFn, type UseBoundStoreWithEqualityFn } from 'zustand/traditional'
 import type { DomEvent, EventManager, PointerCaptureTarget, ThreeEvent } from './events'
 import { calculateDpr, type Camera, isOrthographicCamera, updateCamera } from './utils'
 import type { FixedStage, Stage } from './stages'
@@ -154,15 +155,15 @@ export interface RootState {
   internal: InternalState
 }
 
-export type RootStore = UseBoundStore<StoreApi<RootState>>
+export type RootStore = UseBoundStoreWithEqualityFn<StoreApi<RootState>>
 
 export const context = React.createContext<RootStore>(null!)
 
 export const createStore = (
   invalidate: (state?: RootState, frames?: number) => void,
   advance: (timestamp: number, runGlobalEffects?: boolean, state?: RootState, frame?: XRFrame) => void,
 ): RootStore => {
-  const rootStore = create<RootState>((set, get) => {
+  const rootStore = createWithEqualityFn<RootState>((set, get) => {
     const position = new THREE.Vector3()
     const defaultTarget = new THREE.Vector3()
     const tempTarget = new THREE.Vector3()

packages/fiber/tests/utils.test.ts

@@ -1,5 +1,6 @@
 import * as THREE from 'three'
-import { Instance, RootStore } from '../src'
+import { createWithEqualityFn } from 'zustand/traditional'
+import { type RootStore, type Instance } from '../src'
 import {
   is,
   dispose,
@@ -16,11 +17,7 @@ import {
 } from '../src/core/utils'
 
 // Mocks a Zustand store
-const storeMock = Object.assign(() => null!, {
-  getState: () => null!,
-  setState() {},
-  subscribe: () => () => {},
-}) as unknown as RootStore
+const storeMock = createWithEqualityFn(() => null!) satisfies RootStore
 
 describe('is', () => {
   const myFunc = () => null