fal 1.46.0__py3-none-any.whl → 1.46.1__py3-none-any.whl

fal/_fal_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '1.46.0'
-__version_tuple__ = version_tuple = (1, 46, 0)
+__version__ = version = '1.46.1'
+__version_tuple__ = version_tuple = (1, 46, 1)
 
 __commit_id__ = commit_id = None

fal/distributed/utils.py

@@ -8,7 +8,7 @@ import warnings
 from collections.abc import Callable
 from io import BytesIO
 from pathlib import Path
-from typing import TYPE_CHECKING, Any, Optional, Union
+from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple, Union
 
 if TYPE_CHECKING:
     import torch.multiprocessing as mp
@@ -271,8 +271,8 @@ def wrap_distributed_worker(
     master_port: int,
     timeout: int,
     cwd: Optional[Union[str, Path]],
-    args: tuple[Any],
-    kwargs: dict[str, Any],
+    args: Tuple[Any],
+    kwargs: Dict[str, Any],
 ) -> None:
     """
     Worker function for distributed training or inference.

fal/distributed/worker.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 import inspect
 import os
@@ -9,6 +11,7 @@ import traceback
 import warnings
 from collections.abc import AsyncIterator, Callable, Coroutine
 from concurrent.futures import Future
+from functools import partial
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Optional, Union
 
@@ -70,7 +73,7 @@ class DistributedWorker:
     # Public API
 
     @property
-    def device(self) -> "torch.device":
+    def device(self) -> torch.device:
         """
         :return: The device for the current worker.
         """
@@ -162,6 +165,16 @@ class DistributedWorker:
         self.loop.call_soon_threadsafe(self.loop.stop)
         self.thread.join(timeout=timeout)
 
+    async def _run_sync_in_executor(
+        self,
+        func: Callable[..., Any],
+        *args: Any,
+        **kwargs: Any,
+    ) -> Any:
+        """Run a synchronous function in the executor."""
+        loop = asyncio.get_running_loop()
+        return await loop.run_in_executor(None, partial(func, *args, **kwargs))
+
     def run_in_worker(
         self,
         func: Callable[..., Any],
@@ -174,7 +187,9 @@ class DistributedWorker:
         if inspect.iscoroutinefunction(func):
             coro = func(*args, **kwargs)
         else:
-            coro = asyncio.to_thread(func, *args, **kwargs)
+            # Using in place of asyncio.to_thread
+            # since it's not available in Python 3.8
+            coro = self._run_sync_in_executor(func, *args, **kwargs)
 
         return self.submit(coro)
 
@@ -206,8 +221,8 @@ class DistributedRunner:
     A class to launch and manage distributed workers.
     """
 
-    zmq_socket: Optional["Socket[Any]"]
-    context: Optional["mp.ProcessContext"]
+    zmq_socket: Optional[Socket[Any]]
+    context: Optional[mp.ProcessContext]
     keepalive_timer: Optional[KeepAliveTimer]
 
     def __init__(
@@ -296,7 +311,7 @@ class DistributedRunner:
                 f"Distributed processes are not running. Errors: {self.gather_errors()}"
             )
 
-    def get_zmq_socket(self) -> "Socket[Any]":
+    def get_zmq_socket(self) -> Socket[Any]:
         """
         Returns a ZeroMQ socket of the specified type.
         :param socket_type: The type of the ZeroMQ socket.
@@ -750,7 +765,7 @@ class DistributedRunner:
         assert rank == b"0", "Expected response from worker with rank 0"
         return distributed_deserialize(response)
 
-    async def __aenter__(self) -> "DistributedRunner":
+    async def __aenter__(self) -> DistributedRunner:
         """
         Enter the context manager.
         :return: The DistributedRunner instance.

{fal-1.46.0.dist-info → fal-1.46.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fal
-Version: 1.46.0
+Version: 1.46.1
 Summary: fal is an easy-to-use Serverless Python Framework
 Author: Features & Labels <support@fal.ai>
 Requires-Python: >=3.8
@@ -53,6 +53,7 @@ Requires-Dist: pytest-xdist; extra == "test"
 Requires-Dist: pytest-timeout; extra == "test"
 Requires-Dist: flaky; extra == "test"
 Requires-Dist: boto3; extra == "test"
+Requires-Dist: numpy; extra == "test"
 Provides-Extra: dev
 Requires-Dist: fal[docs,test]; extra == "dev"
 Requires-Dist: openapi-python-client<1,>=0.14.1; extra == "dev"

{fal-1.46.0.dist-info → fal-1.46.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 fal/__init__.py,sha256=wXs1G0gSc7ZK60-bHe-B2m0l_sA6TrFk4BxY0tMoLe8,784
 fal/__main__.py,sha256=4JMK66Wj4uLZTKbF-sT3LAxOsr6buig77PmOkJCRRxw,83
-fal/_fal_version.py,sha256=HSh8n5t3dT12FiqCeiBwPGGXi_9ERts2UyWRnVzpW5U,706
+fal/_fal_version.py,sha256=r7bDyLX7CxvhE0UPuE3CER_HLAjx7gKNT1btlAQda-M,706
 fal/_serialization.py,sha256=npXNsFJ5G7jzBeBIyVMH01Ww34mGY4XWhHpRbSrTtnQ,7598
 fal/_version.py,sha256=1BbTFnucNC_6ldKJ_ZoC722_UkW4S9aDBSW9L0fkKAw,2315
 fal/app.py,sha256=izlqdqfUtNxKW6oJerJYbVaUIZwxlO-DTtqFrcTleyI,28039
@@ -50,8 +50,8 @@ fal/console/__init__.py,sha256=lGPUuTqIM9IKTa1cyyA-MA2iZJKVHp2YydsITZVlb6g,148
 fal/console/icons.py,sha256=De9MfFaSkO2Lqfne13n3PrYfTXJVIzYZVqYn5BWsdrA,108
 fal/console/ux.py,sha256=KMQs3UHQvVHDxDQQqlot-WskVKoMQXOE3jiVkkfmIMY,356
 fal/distributed/__init__.py,sha256=jnLKLzTnF0GurjJnt7xD1-wB-oKWALZcZ1ZK0c6WI4g,134
-fal/distributed/utils.py,sha256=8Wl1WytTtZQw4Ja-95D2vM40P26eYZNZe-fI2KVX7kU,12648
-fal/distributed/worker.py,sha256=TReJBU1P1iXarKXVI8NvlXt7stMWxdvWIozJ2lZTF5w,26704
+fal/distributed/utils.py,sha256=b7zgYg2O9L_kMGup8Mc5x7QRx_gPwwELcmVAXo-oGlU,12661
+fal/distributed/worker.py,sha256=vrxqg4BeGICo_VNvTZWsFPMgGdkZ0WUVt56ljg2T3Zs,27195
 fal/exceptions/__init__.py,sha256=4hq-sy3dMZs6YxvbO_p6R-bK4Tzf7ubvA8AyUR0GVPo,349
 fal/exceptions/_base.py,sha256=PLSOHQs7lftDaRYDHKz9xkB6orQvynmUTi4DrdPnYMs,1797
 fal/exceptions/_cuda.py,sha256=L3qvDNaPTthp95IFSBI6pMt3YbRfn1H0inQkj_7NKF8,1719
@@ -152,8 +152,8 @@ openapi_fal_rest/models/workflow_node_type.py,sha256=-FzyeY2bxcNmizKbJI8joG7byRi
 openapi_fal_rest/models/workflow_schema.py,sha256=4K5gsv9u9pxx2ItkffoyHeNjBBYf6ur5bN4m_zePZNY,2019
 openapi_fal_rest/models/workflow_schema_input.py,sha256=2OkOXWHTNsCXHWS6EGDFzcJKkW5FIap-2gfO233EvZQ,1191
 openapi_fal_rest/models/workflow_schema_output.py,sha256=EblwSPAGfWfYVWw_WSSaBzQVju296is9o28rMBAd0mc,1196
-fal-1.46.0.dist-info/METADATA,sha256=bOq6HL6Lzc-yql9GTNRaTHcqlwMpuNyULT90kR4Rzo8,4185
-fal-1.46.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fal-1.46.0.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
-fal-1.46.0.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
-fal-1.46.0.dist-info/RECORD,,
+fal-1.46.1.dist-info/METADATA,sha256=wv_JLs46I6tnsjgaYQWkgWyAOc4lPNv_QyYdjqz8IFU,4223
+fal-1.46.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fal-1.46.1.dist-info/entry_points.txt,sha256=32zwTUC1U1E7nSTIGCoANQOQ3I7-qHG5wI6gsVz5pNU,37
+fal-1.46.1.dist-info/top_level.txt,sha256=r257X1L57oJL8_lM0tRrfGuXFwm66i1huwQygbpLmHw,21
+fal-1.46.1.dist-info/RECORD,,