fix: need py entrypoint (#431)

BobTheBuidler · Nov 23, 2024 · 78a644c · 78a644c
1 parent 8f8a363
commit 78a644c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/a_sync/a_sync/_kwargs.pyx b/a_sync/a_sync/_kwargs.pyx
@@ -9,6 +9,10 @@ from a_sync.a_sync._flags cimport negate_if_necessary
 from a_sync.a_sync.flags import VIABLE_FLAGS
 
 
+def get_flag_name_legacy(dict kwargs) -> str:
+    return get_flag_name(kwargs)
+
+
 cdef str get_flag_name(dict kwargs):
     """
     Get the name of the flag present in the kwargs.

diff --git a/a_sync/task.py b/a_sync/task.py
@@ -144,7 +144,7 @@ async def process_item(item: int) -> int:
 
         if isinstance(
             wrapped_func, ASyncMethodDescriptor
-        ) and not _kwargs.get_flag_name(wrapped_func_kwargs):
+        ) and not _kwargs.get_flag_name_legacy(wrapped_func_kwargs):
             wrapped_func_kwargs["sync"] = False
         if wrapped_func_kwargs:
             self._wrapped_func_kwargs = wrapped_func_kwargs