From ad5a27239392d02ffec168cdbffc582ee1cb3969 Mon Sep 17 00:00:00 2001
From: Joe Savona <joesavona@fb.com>
Date: Wed, 14 Feb 2024 15:23:37 -0800
Subject: [PATCH] Skip functions using same rules as regular pipeline

---
 .../babel-plugin-react-forget/src/Entrypoint/Program.ts         | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/compiler/packages/babel-plugin-react-forget/src/Entrypoint/Program.ts b/compiler/packages/babel-plugin-react-forget/src/Entrypoint/Program.ts
index 04fa34234a..934ca5fa74 100644
--- a/compiler/packages/babel-plugin-react-forget/src/Entrypoint/Program.ts
+++ b/compiler/packages/babel-plugin-react-forget/src/Entrypoint/Program.ts
@@ -361,7 +361,7 @@ export function compileProgram(
   }
 }
 
-function shouldVisitNode(fn: BabelFn, pass: CompilerPass): boolean {
+export function shouldVisitNode(fn: BabelFn, pass: CompilerPass): boolean {
   if (hasUseMemoCacheCall(fn)) {
     return false;
   }