flyte 0.2.0b8__py3-none-any.whl → 0.2.0b10__py3-none-any.whl

flyte/cli/_gen.py

@@ -1,3 +1,4 @@
+import textwrap
 from os import getcwd
 from typing import Generator, Tuple
 
@@ -9,7 +10,7 @@ import flyte.cli._common as common
 @click.group(name="gen")
 def gen():
     """
-    Generate documentation
+    Generate documentation.
     """
 
 
@@ -18,7 +19,7 @@ def gen():
 @click.pass_obj
 def docs(cfg: common.CLIConfig, doc_type: str, project: str | None = None, domain: str | None = None):
     """
-    Generate documentation
+    Generate documentation.
     """
     if doc_type == "markdown":
         markdown(cfg)
@@ -82,7 +83,7 @@ def markdown(cfg: common.CLIConfig):
         output.append(f"{'#' * (len(cmd_path_parts) + 1)} {cmd_path}")
         if cmd.help:
             output.append("")
-            output.append(f"{cmd.help.strip()}")
+            output.append(f"{dedent(cmd.help)}")
 
         if not cmd.params:
             continue
@@ -109,7 +110,7 @@ def markdown(cfg: common.CLIConfig):
                 if param.default is not None:
                     default_value = f"`{param.default}`"
                     default_value = default_value.replace(f"{getcwd()}/", "")
-                help_text = param.help.strip() if param.help else ""
+                help_text = dedent(param.help) if param.help else ""
                 table_data.append([opts, f"`{param.type.name}`", default_value, help_text])
 
         if not table_data:
@@ -153,3 +154,10 @@ def markdown(cfg: common.CLIConfig):
     print("{{< /grid >}}")
     print()
     print("\n".join(output))
+
+
+def dedent(text: str) -> str:
+    """
+    Remove leading whitespace from a string.
+    """
+    return textwrap.dedent(text).strip("\n")

flyte/cli/_get.py

@@ -11,16 +11,23 @@ from . import _common as common
 @click.group(name="get")
 def get():
     """
-    The `get` subcommand allows you to retrieve various resources from a Flyte deployment.
+    Retrieve resources from a Flyte deployment.
+
+    You can get information about projects, runs, tasks, actions, secrets, logs and input/output values.
 
-    You can get information about projects, runs, tasks, actions, secrets, and more.
     Each command supports optional parameters to filter or specify the resource you want to retrieve.
 
-    Every `get` subcommand for example ``get project` without any arguments will list all projects.
-    `get project my_project` will return the details of the project named `my_project`.
+    Using a `get` subcommand without any arguments will retrieve a list of available resources to get.
+    For example:
+
+    * `get project` (without specifiying aproject), will list all projects.
+    * `get project my_project` will return the details of the project named `my_project`.
+
+    In some cases, a partially specified command will act as a filter and return available further parameters.
+    For example:
 
-    In some cases `get action my_run` will return all actions for the run named `my_run` and
-    `get action my_run my_action` will return the details of the action named `my_action` for the run `my_run`.
+    * `get action my_run` will return all actions for the run named `my_run`.
+    * `get action my_run my_action` will return the details of the action named `my_action` for the run `my_run`.
     """
 
 
@@ -29,7 +36,7 @@ def get():
 @click.pass_obj
 def project(cfg: common.CLIConfig, name: str | None = None):
     """
-    Retrieve a list of all projects or details of a specific project by name.
+    Get a list of all projects, or details of a specific project by name.
     """
     from flyte.remote import Project
 
@@ -48,7 +55,7 @@ def project(cfg: common.CLIConfig, name: str | None = None):
 @click.pass_obj
 def run(cfg: common.CLIConfig, name: str | None = None, project: str | None = None, domain: str | None = None):
     """
-    Get list of all runs or details of a specific run by name.
+    Get a list of all runs, or details of a specific run by name.
 
     The run details will include information about the run, its status, but only the root action will be shown.
 
@@ -78,9 +85,9 @@ def task(
     domain: str | None = None,
 ):
     """
-    Retrieve a list of all tasks or details of a specific task by name and version.
+    Retrieve a list of all tasks, or details of a specific task by name and version.
 
-    Currently name+version are required to get a specific task.
+    Currently, both `name` and `version` are required to get a specific task.
     """
     from flyte.remote import Task
 
@@ -156,23 +163,24 @@ def logs(
 ):
     """
     Stream logs for the provided run or action.
-    If only the run is provided, only the logs for the parent action will be streamed.
+    If only the run is provided, only the logs for the parent action will be streamed:
 
-    Example:
     ```bash
-    flyte get logs my_run
+    $ flyte get logs my_run
     ```
 
-    But, if you want to see the logs for a specific action, you can provide the action name as well:
+    If you want to see the logs for a specific action, you can provide the action name as well:
+
     ```bash
-    flyte get logs my_run my_action
+    $ flyte get logs my_run my_action
     ```
-    By default logs will be shown in the raw format, will scroll on the terminal. If automatic scrolling and only
-    tailing --lines lines is desired, use the `--pretty` flag:
+
+    By default, logs will be shown in the raw format and will scroll the terminal.
+    If automatic scrolling and only tailing `--lines` number of lines is desired, use the `--pretty` flag:
+
     ```bash
-    flyte get logs my_run my_action --pretty --lines 50
+    $ flyte get logs my_run my_action --pretty --lines 50
     ```
-
     """
     import flyte.remote as remote
 
@@ -206,7 +214,7 @@ def secret(
     domain: str | None = None,
 ):
     """
-    Retrieve a list of all secrets or details of a specific secret by name.
+    Get a list of all secrets, or details of a specific secret by name.
     """
     import flyte.remote as remote
 
@@ -236,18 +244,18 @@ def io(
 ):
     """
     Get the inputs and outputs of a run or action.
-    if only the run name is provided, it will show the inputs and outputs of the root action of that run.
+    If only the run name is provided, it will show the inputs and outputs of the root action of that run.
     If an action name is provided, it will show the inputs and outputs for that action.
-
     If `--inputs-only` or `--outputs-only` is specified, it will only show the inputs or outputs respectively.
 
-    Example:
+    Examples:
+
     ```bash
-    flyte get io my_run
+    $ flyte get io my_run
     ```
-    or
+
     ```bash
-    flyte get io my_run my_action
+    $ flyte get io my_run my_action
     ```
     """
     if inputs_only and outputs_only:
@@ -293,7 +301,7 @@ def io(
 @click.pass_obj
 def config(cfg: common.CLIConfig):
     """
-    Shows the automatically detected configuration to connect with remote Flyte services.
+    Shows the automatically detected configuration to connect with the remote backend.
 
     The configuration will include the endpoint, organization, and other settings that are used by the CLI.
     """

flyte/cli/_params.py

@@ -23,7 +23,7 @@ from mashumaro.codecs.json import JSONEncoder
 
 from flyte._logging import logger
 from flyte.io import Dir, File
-from flyte.io.pickle.transformer import FlytePickleTransformer
+from flyte.types._pickle import FlytePickleTransformer
 
 
 class StructuredDataset:

flyte/cli/main.py

@@ -13,7 +13,7 @@ from ._run import run
 click.rich_click.COMMAND_GROUPS = {
     "flyte": [
         {
-            "name": "Running Workflows",
+            "name": "Running workflows",
             "commands": ["run", "abort"],
         },
         {
@@ -21,7 +21,7 @@ click.rich_click.COMMAND_GROUPS = {
             "commands": ["create", "deploy", "get"],
         },
         {
-            "name": "Documentation Generation",
+            "name": "Documentation generation",
             "commands": ["gen"],
         },
     ]
@@ -58,13 +58,13 @@ def _verbosity_to_loglevel(verbosity: int) -> int | None:
     "--endpoint",
     type=str,
     required=False,
-    help="The endpoint to connect to, this will override any config and simply used pkce to connect.",
+    help="The endpoint to connect to. This will override any configuration file and simply use `pkce` to connect.",
 )
 @click.option(
-    "--insecure/--secure",
+    "--insecure",
     is_flag=True,
     required=False,
-    help="Use insecure connection to the endpoint. If secure is specified, the CLI will use TLS.",
+    help="Use an insecure connection to the endpoint. If not specified, the CLI will use TLS.",
     type=bool,
     default=None,
     show_default=True,
@@ -73,7 +73,7 @@ def _verbosity_to_loglevel(verbosity: int) -> int | None:
     "-v",
     "--verbose",
     required=False,
-    help="Show verbose messages and exception traces, multiple times increases verbosity (e.g., -vvv).",
+    help="Show verbose messages and exception traces. Repeating multiple times increases the verbosity (e.g., -vvv).",
     count=True,
     default=0,
     type=int,
@@ -82,7 +82,7 @@ def _verbosity_to_loglevel(verbosity: int) -> int | None:
     "--org",
     type=str,
     required=False,
-    help="Organization to use",
+    help="The organization to which the command applies.",
 )
 @click.option(
     "-c",
@@ -90,8 +90,7 @@ def _verbosity_to_loglevel(verbosity: int) -> int | None:
     "config_file",
     required=False,
     type=click.Path(exists=True),
-    help="Path to config file (YAML format) to use for the CLI. If not specified,"
-    " the default config file will be used.",
+    help="Path to the configuration file to use. If not specified, the default configuration file is used.",
 )
 @click.rich_config(help_config=help_config)
 @click.pass_context
@@ -104,34 +103,38 @@ def main(
     config_file: str | None,
 ):
     """
-    ### Flyte entrypoint for the CLI
-    The Flyte CLI is a command line interface for interacting with Flyte.
+    The Flyte CLI is the the command line interface for working with the Flyte SDK and backend.
 
-    The flyte cli follows a simple verb based structure, where the top-level commands are verbs that describe the action
-    to be taken, and the subcommands are nouns that describe the object of the action.
+    It follows a simple verb/noun structure,
+    where the top-level commands are verbs that describe the action to be taken,
+    and the subcommands are nouns that describe the object of the action.
 
-    The root command can be used to configure the CLI for most commands, such as setting the endpoint,
-     organization, and verbosity level.
+    The root command can be used to configure the CLI for persistent settings,
+    such as the endpoint, organization, and verbosity level.
 
-     Example: Set endpoint and organization
-     ```bash
-      flyte --endpoint <endpoint> --org <org> get project <project_name>
-     ```
+    Set endpoint and organization:
 
-     Example: Increase verbosity level (This is useful for debugging, this will show more logs and exception traces)
-      ```bash
-      flyte -vvv get logs <run-name>
-      ```
+    ```bash
+    $ flyte --endpoint <endpoint> --org <org> get project <project_name>
+    ```
+
+    Increase verbosity level (This is useful for debugging,
+    this will show more logs and exception traces):
+
+    ```bash
+    $ flyte -vvv get logs <run-name>
+    ```
+
+    Override the default config file:
 
-      Example: Override the default config file
     ```bash
-    flyte --config /path/to/config.yaml run ...
+    $ flyte --config /path/to/config.yaml run ...
     ```
 
-    👉 [Documentation](https://www.union.ai/docs/flyte/user-guide/) \n
-    👉 [GitHub](https://github.com/flyteorg/flyte) - Please leave a ⭐. \n
-    👉 [Slack](https://slack.flyte.org) - Join the community and ask questions.
-    👉 [Issues](https://github.com/flyteorg/flyte/issues)
+    * [Documentation](https://www.union.ai/docs/flyte/user-guide/)
+    * [GitHub](https://github.com/flyteorg/flyte): Please leave a star if you like Flyte!
+    * [Slack](https://slack.flyte.org): Join the community and ask questions.
+    * [Issues](https://github.com/flyteorg/flyte/issues)
 
     """
     import flyte.config as config

flyte/extras/_container.py

@@ -12,17 +12,14 @@ from flyte.models import NativeInterface, SerializationContext
 _PRIMARY_CONTAINER_NAME_FIELD = "primary_container_name"
 
 
-def _extract_command_key(cmd: str, **kwargs) -> Any:
+def _extract_command_key(cmd: str, **kwargs) -> List[Any] | None:
     """
     Extract the key from the command using regex.
     """
     import re
 
-    input_regex = r"^\{\{\s*\.inputs\.(.*?)\s*\}\}$"
-    match = re.match(input_regex, cmd)
-    if match:
-        return match.group(1)
-    return None
+    input_regex = r"\{\{\.inputs\.([a-zA-Z0-9_]+)\}\}"
+    return re.findall(input_regex, cmd)
 
 
 def _extract_path_command_key(cmd: str, input_data_dir: Optional[str]) -> Optional[str]:
@@ -70,7 +67,7 @@ class ContainerTask(TaskTemplate):
         input_data_dir: str | pathlib.Path = "/var/inputs",
         output_data_dir: str | pathlib.Path = "/var/outputs",
         metadata_format: MetadataFormat = "JSON",
-        local_logs: bool = False,
+        local_logs: bool = True,
         **kwargs,
     ):
         super().__init__(
@@ -106,34 +103,33 @@ class ContainerTask(TaskTemplate):
         For FlyteFile and FlyteDirectory commands, e.g., "/var/inputs/inputs", we extract the key from strings that
          begin with the specified `input_data_dir`.
         """
-        # from flytekit.types.directory import FlyteDirectory
-        # from flytekit.types.file import FlyteFile
+        from flyte.io import Dir, File
 
         volume_binding: Dict[str, Dict[str, str]] = {}
         path_k = _extract_path_command_key(cmd, str(self._input_data_dir))
-        k = path_k if path_k else _extract_command_key(cmd)
-
-        if k:
-            input_val = kwargs.get(k)
-            # TODO: Add support file and directory transformer first
-            # if type(input_val) in [FlyteFile, FlyteDirectory]:
-            #     if not path_k:
-            #         raise AssertionError(
-            #             "FlyteFile and FlyteDirectory commands should not use the template syntax like this:
-            #             {{.inputs.infile}}\n"
-            #             "Please use a path-like syntax, such as: /var/inputs/infile.\n"
-            #             "This requirement is due to how Flyte Propeller processes template syntax inputs."
-            #         )
-            #     local_flyte_file_or_dir_path = str(input_val)
-            #     remote_flyte_file_or_dir_path = os.path.join(self._input_data_dir, k)  # type: ignore
-            #     volume_binding[local_flyte_file_or_dir_path] = {
-            #         "bind": remote_flyte_file_or_dir_path,
-            #         "mode": "rw",
-            #     }
-            #     command = remote_flyte_file_or_dir_path
-            command = str(input_val)
-        else:
-            command = cmd
+        keys = path_k if path_k else _extract_command_key(cmd)
+
+        if keys:
+            for k in keys:
+                input_val = kwargs.get(k)
+                # TODO: Add support file and directory transformer first
+                if type(input_val) in [File, Dir]:
+                    if not path_k:
+                        raise AssertionError(
+                            "File and Directory commands should not use the template syntax "
+                            "like this: {{.inputs.infile}}\n"
+                            "Please use a path-like syntax, such as: /var/inputs/infile.\n"
+                            "This requirement is due to how Flyte Propeller processes template syntax inputs."
+                        )
+                    local_flyte_file_or_dir_path = str(input_val)
+                    remote_flyte_file_or_dir_path = os.path.join(self._input_data_dir, k)  # type: ignore
+                    volume_binding[local_flyte_file_or_dir_path] = {
+                        "bind": remote_flyte_file_or_dir_path,
+                        "mode": "rw",
+                    }
+                    command = remote_flyte_file_or_dir_path
+                else:
+                    command = cmd
 
         return command, volume_binding
 
@@ -235,6 +231,7 @@ class ContainerTask(TaskTemplate):
             raise AssertionError(f"Only Image objects are supported, not strings. Got {self._image} instead.")
         uri = self._image.uri
         self._pull_image_if_not_exists(client, uri)
+        print(f"Command: {commands!r}")
 
         container = client.containers.run(uri, command=commands, remove=True, volumes=volume_bindings, detach=True)
 

flyte/io/__init__.py

@@ -3,9 +3,25 @@
 
 This package contains additional data types beyond the primitive data types in python to abstract data flow
 of large datasets in Union.
+
 """
 
-__all__ = ["Dir", "File"]
+__all__ = [
+    "Dir",
+    "File",
+    "StructuredDataset",
+    "StructuredDatasetDecoder",
+    "StructuredDatasetEncoder",
+    "StructuredDatasetTransformerEngine",
+    "lazy_import_structured_dataset_handler",
+]
 
 from ._dir import Dir
 from ._file import File
+from ._structured_dataset import (
+    StructuredDataset,
+    StructuredDatasetDecoder,
+    StructuredDatasetEncoder,
+    StructuredDatasetTransformerEngine,
+    lazy_import_structured_dataset_handler,
+)

flyte/io/_file.py

@@ -232,6 +232,8 @@ class File(BaseModel, Generic[T], SerializableType):
             # This code is broadly similar to what storage.get_stream does, but without actually reading from the stream
             file_handle = None
             try:
+                if "b" not in mode:
+                    raise ValueError("Mode must include 'b' for binary access, when using remote files.")
                 if isinstance(fs, AsyncFileSystem):
                     file_handle = await fs.open_async(self.path, mode)
                     yield file_handle

flyte/io/{structured_dataset → _structured_dataset}/basic_dfs.py

@@ -9,7 +9,7 @@ from fsspec.core import split_protocol, strip_protocol
 import flyte.storage as storage
 from flyte._logging import logger
 from flyte._utils import lazy_module
-from flyte.io.structured_dataset.structured_dataset import (
+from flyte.io._structured_dataset.structured_dataset import (
     CSV,
     PARQUET,
     StructuredDataset,

flyte/io/{structured_dataset → _structured_dataset}/structured_dataset.py

@@ -168,7 +168,7 @@ class StructuredDataset(SerializableType, DataClassJSONMixin):
         return self._literal_sd
 
     def open(self, dataframe_type: Type[DF]):
-        from flyte.io.structured_dataset import lazy_import_structured_dataset_handler
+        from flyte.io._structured_dataset import lazy_import_structured_dataset_handler
 
         """
         Load the handler if needed. For the use case like:

flyte/models.py

@@ -5,7 +5,9 @@ import os
 import pathlib
 import tempfile
 from dataclasses import dataclass, field, replace
-from typing import TYPE_CHECKING, Any, Callable, Dict, Optional, Tuple, Type
+from typing import TYPE_CHECKING, Any, Callable, Dict, Literal, Optional, Tuple, Type
+
+import rich.repr
 
 from flyte._docstring import Docstring
 from flyte._interface import extract_return_annotation
@@ -27,6 +29,7 @@ def generate_random_name() -> str:
     return str(uuid4())  # Placeholder for actual random name generation logic
 
 
+@rich.repr.auto
 @dataclass(frozen=True, kw_only=True)
 class ActionID:
     """
@@ -68,6 +71,7 @@ class ActionID:
         return self.new_sub_action(new_name)
 
 
+@rich.repr.auto
 @dataclass(frozen=True, kw_only=True)
 class RawDataPath:
     """
@@ -91,6 +95,7 @@ class RawDataPath:
                 # Create a temporary directory for data storage
                 p = tempfile.mkdtemp()
                 logger.debug(f"Creating temporary directory for data storage: {p}")
+                pathlib.Path(p).mkdir(parents=True, exist_ok=True)
                 return RawDataPath(path=p)
             case str():
                 return RawDataPath(path=local_folder)
@@ -133,11 +138,13 @@ class RawDataPath:
         return remote_path
 
 
+@rich.repr.auto
 @dataclass(frozen=True)
 class GroupData:
     name: str
 
 
+@rich.repr.auto
 @dataclass(frozen=True, kw_only=True)
 class TaskContext:
     """
@@ -160,6 +167,7 @@ class TaskContext:
     code_bundle: CodeBundle | None = None
     compiled_image_cache: ImageCache | None = None
     data: Dict[str, Any] = field(default_factory=dict)
+    mode: Literal["local", "remote", "hybrid"] = "remote"
 
     def replace(self, **kwargs) -> TaskContext:
         if "data" in kwargs:
@@ -177,6 +185,7 @@ class TaskContext:
         return self.data.get(key)
 
 
+@rich.repr.auto
 @dataclass(frozen=True, kw_only=True)
 class CodeBundle:
     """
@@ -211,6 +220,7 @@ class CodeBundle:
         return replace(self, downloaded_path=path)
 
 
+@rich.repr.auto
 @dataclass(frozen=True)
 class Checkpoints:
     """

flyte/syncify/_api.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import asyncio
 import atexit
 import concurrent.futures
+import functools
 import inspect
 import threading
 from typing import (
@@ -14,7 +15,6 @@ from typing import (
     Iterator,
     ParamSpec,
     Protocol,
-    Type,
     TypeVar,
     Union,
     cast,
@@ -92,7 +92,9 @@ class _BackgroundLoop:
 
     def iterate_in_loop_sync(self, async_gen: AsyncIterator[R_co]) -> Iterator[R_co]:
         # Create an iterator that pulls items from the async generator
-        assert self.thread.name != threading.current_thread().name, "Cannot run coroutine in the same thread"
+        assert self.thread.name != threading.current_thread().name, (
+            f"Cannot run coroutine in the same thread {self.thread.name}"
+        )
         while True:
             try:
                 # use __anext__() and cast to Coroutine so mypy is happy
@@ -104,12 +106,19 @@ class _BackgroundLoop:
             except (StopAsyncIteration, StopIteration):
                 break
 
-    def call_in_loop_sync(self, coro: Coroutine[Any, Any, R_co]) -> R_co:
+    def call_in_loop_sync(self, coro: Coroutine[Any, Any, R_co]) -> R_co | Iterator[R_co]:
         """
         Run the given coroutine in the background loop and return its result.
         """
-        future: concurrent.futures.Future[R_co] = asyncio.run_coroutine_threadsafe(coro, self.loop)
-        return future.result()
+        future: concurrent.futures.Future[R_co | AsyncIterator[R_co]] = asyncio.run_coroutine_threadsafe(
+            coro, self.loop
+        )
+        result = future.result()
+        if result is not None and hasattr(result, "__aiter__"):
+            # If the result is an async iterator, we need to convert it to a sync iterator
+            return cast(Iterator[R_co], self.iterate_in_loop_sync(cast(AsyncIterator[R_co], result)))
+        # Otherwise, just return the result
+        return result
 
     async def iterate_in_loop(self, async_gen: AsyncIterator[R_co]) -> AsyncIterator[R_co]:
         """
@@ -131,7 +140,9 @@ class _BackgroundLoop:
                 # Wrap the future in an asyncio Future to yield it in an async context
                 aio_future: asyncio.Future[R_co] = asyncio.wrap_future(future)
                 # await for the future to complete and yield its result
-                yield await aio_future
+                v = await aio_future
+                print(f"Yielding value: {v}")
+                yield v
             except StopAsyncIteration:
                 break
 
@@ -159,8 +170,6 @@ class _SyncWrapper:
         self,
         fn: Any,
         bg_loop: _BackgroundLoop,
-        instance: Any = None,
-        owner: Type | None = None,
         underlying_obj: Any = None,
     ):
         self.fn = fn
@@ -168,6 +177,12 @@ class _SyncWrapper:
         self._underlying_obj = underlying_obj
 
     def __call__(self, *args: Any, **kwargs: Any) -> Any:
+        if threading.current_thread().name == self._bg_loop.thread.name:
+            # If we are already in the background loop thread, we can call the function directly
+            raise AssertionError(
+                f"Deadlock detected: blocking call used in syncify thread {self._bg_loop.thread.name} "
+                f"when calling function {self.fn}, use .aio() if in an async call."
+            )
         # bind method if needed
         coro_fn = self.fn
 
@@ -194,7 +209,9 @@ class _SyncWrapper:
             # If we have an owner, we need to bind the method to the owner (for classmethods or staticmethods)
             fn = self._underlying_obj.__get__(None, owner)
 
-        return _SyncWrapper(fn, bg_loop=self._bg_loop, underlying_obj=self._underlying_obj)
+        wrapper = _SyncWrapper(fn, bg_loop=self._bg_loop, underlying_obj=self._underlying_obj)
+        functools.update_wrapper(wrapper, self.fn)
+        return wrapper
 
     def aio(self, *args: Any, **kwargs: Any) -> Any:
         fn = self.fn
@@ -262,15 +279,26 @@ class Syncify:
 
     def __call__(self, obj):
         if isinstance(obj, classmethod):
-            return _SyncWrapper(obj.__func__, bg_loop=self._bg_loop, underlying_obj=obj)
+            wrapper = _SyncWrapper(obj.__func__, bg_loop=self._bg_loop, underlying_obj=obj)
+            functools.update_wrapper(wrapper, obj.__func__)
+            return wrapper
+
         if isinstance(obj, staticmethod):
-            return staticmethod(cast(Any, _SyncWrapper(obj.__func__, bg_loop=self._bg_loop)))
+            fn = obj.__func__
+            wrapper = _SyncWrapper(fn, bg_loop=self._bg_loop)
+            functools.update_wrapper(wrapper, fn)
+            return staticmethod(wrapper)
+
         if inspect.isasyncgenfunction(obj):
-            # If the function is an async generator, we need to handle it differently
-            return cast(Callable[P, Iterator[R_co]], _SyncWrapper(obj, bg_loop=self._bg_loop))
+            wrapper = _SyncWrapper(obj, bg_loop=self._bg_loop)
+            functools.update_wrapper(wrapper, obj)
+            return cast(Callable[P, Iterator[R_co]], wrapper)
+
         if inspect.iscoroutinefunction(obj):
-            # If the function is a coroutine, we can wrap it directly
-            return _SyncWrapper(obj, bg_loop=self._bg_loop)
+            wrapper = _SyncWrapper(obj, bg_loop=self._bg_loop)
+            functools.update_wrapper(wrapper, obj)
+            return wrapper
+
         raise TypeError(
             "Syncify can only be applied to async functions, async generators, async classmethods or staticmethods."
         )