Update FuncX to >=1.0.3 (#17)

psbench/benchmarks/funcx_tasks/main.py

@@ -26,21 +26,11 @@
 from psbench.logging import TESTING_LOG_LEVEL
 from psbench.proxystore import init_store_from_args
 from psbench.tasks.pong import pong
+from psbench.tasks.pong import pong_proxy
 from psbench.utils import randbytes
 
-logger = logging.getLogger('funcx-test')
-
-
-def pong_proxy(*args, **kwargs):  # type: ignore
-    """Trampoline function for psbench.tasks.pong.pong_proxy.
 
-    The trampoline is needed to remove type annotations since there is an
-    issue in FuncX with serializing functions with type annotations.
-    (See FuncX #901)
-    """
-    from psbench.tasks.pong import pong_proxy as _pong_proxy
-
-    return _pong_proxy(*args, **kwargs)
+logger = logging.getLogger('funcx-test')
 
 
 @dataclasses.dataclass

psbench/tasks/pong.py

@@ -66,6 +66,7 @@ def pong_proxy(
     from proxystore.store import get_store
     from proxystore.store import UnknownStoreError
 
+    from psbench.tasks.pong import ProxyStats
     from psbench.utils import randbytes
 
     assert isinstance(data, Proxy)

setup.cfg

@@ -18,8 +18,8 @@ classifiers =
 [options]
 packages = find:
 install_requires =
-    funcx==1.0.*
-    funcx-endpoint==1.0.*
+    funcx>=1.0.3
+    funcx-endpoint>=1.0.3
     proxystore[endpoints]==0.4.0a2
     requests
 python_requires = >=3.7