fal 1.46.0__tar.gz → 1.46.1__tar.gz

{fal-1.46.0/fal.egg-info → fal-1.46.1}/PKG-INFO

@@ -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 → fal-1.46.1/fal.egg-info}/PKG-INFO

@@ -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 → fal-1.46.1}/fal.egg-info/requires.txt

@@ -54,3 +54,4 @@ pytest-xdist
 pytest-timeout
 flaky
 boto3
+numpy

{fal-1.46.0 → fal-1.46.1}/pyproject.toml

@@ -77,6 +77,7 @@ test = [
     "pytest-timeout",
     "flaky",
     "boto3",
+    "numpy",
 ]
 dev = [
     "fal[docs,test]",

{fal-1.46.0 → fal-1.46.1}/src/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-1.46.0 → fal-1.46.1}/src/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-1.46.0 → fal-1.46.1}/src/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.