fix(decorator): reject wrapper as dunder methods (#5030)

fix(decorator): gated wrapper to not support dunder methods Signed-off-by: Aaron Pham <contact@aarnphm.xyz>
bentoml · Oct 17, 2024 · b696903 · b696903
1 parent d8b6ef8
commit b696903
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/_bentoml_sdk/decorators.py b/src/_bentoml_sdk/decorators.py
@@ -82,6 +82,8 @@ def api(
     """
 
     def wrapper(func: t.Callable[t.Concatenate[t.Any, P], R]) -> APIMethod[P, R]:
+        if func.__name__.startswith("__"):
+            raise ValueError("API methods cannot start with '__'")
         params: dict[str, t.Any] = {
             "batchable": batchable,
             "batch_dim": batch_dim,
@@ -182,6 +184,8 @@ def task(
     """
 
     def wrapper(func: t.Callable[t.Concatenate[t.Any, P], R]) -> APIMethod[P, R]:
+        if func.__name__.startswith("__"):
+            raise ValueError("API methods cannot start with '__'")
         params: dict[str, t.Any] = {
             "batchable": batchable,
             "batch_dim": batch_dim,