Fix Modal deprecation of f.call(...) (#320)

* Fix Modal deprecation of `f.call(...)` * Allow redefinition for mypy
cubed-dev · Nov 10, 2023 · 19af1dd · 19af1dd
1 parent 339b5a0
commit 19af1dd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/cubed/runtime/executors/modal_async.py b/cubed/runtime/executors/modal_async.py
@@ -60,15 +60,15 @@ async def map_unordered(
 
     def create_futures_func(input, **kwargs):
         return [
-            (i, asyncio.ensure_future(app_function.call.aio(i, **kwargs)))
+            (i, asyncio.ensure_future(app_function.remote.aio(i, **kwargs)))
             for i in input
         ]
 
     backup_function = backup_function or app_function
 
     def create_backup_futures_func(input, **kwargs):
         return [
-            (i, asyncio.ensure_future(backup_function.call.aio(i, **kwargs)))
+            (i, asyncio.ensure_future(backup_function.remote.aio(i, **kwargs)))
             for i in input
         ]
 

diff --git a/setup.cfg b/setup.cfg
@@ -15,6 +15,7 @@ ignore =
 exclude = tests|doc
 files = .
 show_error_codes = True
+allow_redefinition = True
 
 # Most of the numerical computing stack doesn't have type annotations yet.
 [mypy-apache_beam.*]