fix: slightly incorrect check for objects belonging to ivy

ivy-llc · Jan 16, 2025 · 8bf39fc · 8bf39fc
1 parent 026cee8
commit 8bf39fc
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ivy/transpiler/utils/inspect_utils.py b/ivy/transpiler/utils/inspect_utils.py
@@ -172,7 +172,11 @@ def _validate_object(object: Union[FunctionType, MethodType, type], source: str)
         )
 
     # Check if the object belongs to the Ivy framework
-    if hasattr(object, "__module__") and object.__module__.startswith("ivy."):
+    if (
+        hasattr(object, "__module__") and
+        object.__module__.startswith("ivy.") and
+        "ivy.transpiler" not in object.__module__
+    ):
         if source == "ivy":
             return object
         raise InvalidSourceException(