MAINT Remove wrapt dependency

dev-requirements.txt

@@ -572,7 +572,6 @@ wrapt==1.15.0
     # via
     #   aiobotocore
     #   deprecated
-    #   flytekit
     #   tensorflow
 yarl==1.9.2
     # via aiohttp

doc-requirements.txt

@@ -1449,7 +1449,6 @@ wrapt==1.15.0
     #   aiobotocore
     #   astroid
     #   deprecated
-    #   flytekit
     #   pandera
     #   tensorflow
 xarray==2023.8.0

flytekit/exceptions/scopes.py

@@ -1,8 +1,7 @@
+from functools import wraps as _wraps
 from sys import exc_info as _exc_info
 from traceback import format_tb as _format_tb
 
-from wrapt import decorator as _decorator
-
 from flytekit.exceptions import base as _base_exceptions
 from flytekit.exceptions import system as _system_exceptions
 from flytekit.exceptions import user as _user_exceptions
@@ -134,8 +133,22 @@ def _is_base_context():
     return _CONTEXT_STACK[-2] == _NULL_CONTEXT
 
 
+def _decorator(outer_f):
+    """Decorate a function with signature func(wrapped, args, kwargs)."""
+
+    @_wraps(outer_f)
+    def inner_decorator(inner_f):
+        @_wraps(inner_f)
+        def f(*args, **kwargs):
+            return outer_f(inner_f, args, kwargs)
+
+        return f
+
+    return inner_decorator
+
+
 @_decorator
-def system_entry_point(wrapped, instance, args, kwargs):
+def system_entry_point(wrapped, args, kwargs):
     """
     The reason these two (see the user one below) decorators exist is to categorize non-Flyte exceptions at arbitrary
     locations. For example, while there is a separate ecosystem of Flyte-defined user and system exceptions
@@ -178,7 +191,7 @@ def system_entry_point(wrapped, instance, args, kwargs):
 
 
 @_decorator
-def user_entry_point(wrapped, instance, args, kwargs):
+def user_entry_point(wrapped, args, kwargs):
     """
     See the comment for the system_entry_point above as well.
 

setup.py

@@ -77,7 +77,6 @@
         "typing_extensions",
         "urllib3>=1.22,<2.0.0",
         "wheel>=0.30.0,<1.0.0",
-        "wrapt>=1.0.0,<2.0.0",
     ],
     extras_require=extras_require,
     scripts=[

tests/flytekit/integration/remote/mock_flyte_repo/workflows/requirements.txt

@@ -369,7 +369,6 @@ wrapt==1.15.0
     # via
     #   aiobotocore
     #   deprecated
-    #   flytekit
 yarl==1.9.2
     # via aiohttp
 zipp==3.16.2