prefect-client 3.3.6.dev4__tar.gz → 3.3.7.dev1__tar.gz

{prefect_client-3.3.6.dev4 → prefect_client-3.3.7.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect-client
-Version: 3.3.6.dev4
+Version: 3.3.7.dev1
 Summary: Workflow orchestration and management.
 Project-URL: Changelog, https://github.com/PrefectHQ/prefect/releases
 Project-URL: Documentation, https://docs.prefect.io

prefect_client-3.3.7.dev1/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.3.7.dev1"
+__build_date__ = "2025-04-26 08:07:02.330443+00:00"
+__git_commit__ = "cd10dc7dbb1bf043aeec4d2b21cd3e68742f474a"
+__dirty__ = False

{prefect_client-3.3.6.dev4 → prefect_client-3.3.7.dev1}/src/prefect/_versioning.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import os
 from enum import Enum
 from typing import Any, Callable, Coroutine, Dict, Literal, Optional
+from urllib.parse import urlparse
 
 from anyio import run_process
 from pydantic import Field, model_validator
@@ -49,7 +50,7 @@ async def get_github_version_info(
 
     Args:
         version: The commit SHA, falls back to GITHUB_SHA env var
-        branch: The git branch, falls back to GITHUB_REF env var
+        branch: The git branch, falls back to GITHUB_REF_NAME env var
         repository: The repository name, falls back to GITHUB_REPOSITORY env var
         url: The repository URL, constructed from GITHUB_SERVER_URL/GITHUB_REPOSITORY if not provided
 
@@ -60,14 +61,16 @@ async def get_github_version_info(
         ValueError: If any required fields cannot be determined
     """
     version = version or os.getenv("GITHUB_SHA")
-    branch = branch or os.getenv("GITHUB_REF")
+    branch = branch or os.getenv("GITHUB_REF_NAME")
     repository = repository or os.getenv("GITHUB_REPOSITORY")
     url = url or f"{os.getenv('GITHUB_SERVER_URL')}/{repository}"
 
     if not version:
         raise ValueError("version is required - must be provided or set in GITHUB_SHA")
     if not branch:
-        raise ValueError("branch is required - must be provided or set in GITHUB_REF")
+        raise ValueError(
+            "branch is required - must be provided or set in GITHUB_REF_NAME"
+        )
     if not repository:
         raise ValueError(
             "repository is required - must be provided or set in GITHUB_REPOSITORY"
@@ -104,8 +107,10 @@ async def get_git_version_info(
             remote_url = result.stdout.decode().strip()
 
             # Extract just the repository name (last part of the path)
-            repo_name = os.path.basename(remote_url.split(":")[1].rstrip(".git"))
-            repository = repo_name
+            repo_url = urlparse(remote_url)
+            repository = repo_url.path.strip("/")
+            if repository.endswith(".git"):
+                repository = repository[:-4]
 
         if not url and repository:
             # Use the full remote URL as the URL

{prefect_client-3.3.6.dev4 → prefect_client-3.3.7.dev1}/src/prefect/flows.py

@@ -65,7 +65,7 @@ from prefect.exceptions import (
     UnspecifiedFlowError,
 )
 from prefect.filesystems import LocalFileSystem, ReadableDeploymentStorage
-from prefect.futures import PrefectFuture
+from prefect.futures import PrefectFlowRunFuture, PrefectFuture
 from prefect.logging import get_logger
 from prefect.logging.loggers import flow_run_logger
 from prefect.results import ResultSerializer, ResultStorage
@@ -2104,6 +2104,106 @@ class InfrastructureBoundFlow(Flow[P, R]):
                 )
             )
 
+    def submit(self, *args: P.args, **kwargs: P.kwargs) -> PrefectFlowRunFuture[R]:
+        """
+        EXPERIMENTAL: This method is experimental and may be removed or changed in future
+            releases.
+
+        Submit the flow to run on remote infrastructure.
+
+        Args:
+            *args: Positional arguments to pass to the flow.
+            **kwargs: Keyword arguments to pass to the flow.
+
+        Returns:
+            A `PrefectFlowRunFuture` that can be used to retrieve the result of the flow run.
+
+        Examples:
+            Submit a flow to run on Kubernetes:
+
+            ```python
+            from prefect import flow
+            from prefect_kubernetes.experimental import kubernetes
+
+            @kubernetes(work_pool="my-kubernetes-work-pool")
+            @flow
+            def my_flow(x: int, y: int):
+                return x + y
+
+            future = my_flow.submit(x=1, y=2)
+            result = future.result()
+            print(result)
+            ```
+        """
+
+        async def submit_func():
+            async with self.worker_cls(work_pool_name=self.work_pool) as worker:
+                parameters = get_call_parameters(self, args, kwargs)
+                return await worker.submit(
+                    flow=self,
+                    parameters=parameters,
+                    job_variables=self.job_variables,
+                )
+
+        return run_coro_as_sync(submit_func())
+
+    def with_options(
+        self,
+        *,
+        name: Optional[str] = None,
+        version: Optional[str] = None,
+        retries: Optional[int] = None,
+        retry_delay_seconds: Optional[Union[int, float]] = None,
+        description: Optional[str] = None,
+        flow_run_name: Optional[Union[Callable[[], str], str]] = None,
+        task_runner: Union[
+            Type[TaskRunner[PrefectFuture[Any]]], TaskRunner[PrefectFuture[Any]], None
+        ] = None,
+        timeout_seconds: Union[int, float, None] = None,
+        validate_parameters: Optional[bool] = None,
+        persist_result: Optional[bool] = NotSet,  # type: ignore
+        result_storage: Optional[ResultStorage] = NotSet,  # type: ignore
+        result_serializer: Optional[ResultSerializer] = NotSet,  # type: ignore
+        cache_result_in_memory: Optional[bool] = None,
+        log_prints: Optional[bool] = NotSet,  # type: ignore
+        on_completion: Optional[list[FlowStateHook[P, R]]] = None,
+        on_failure: Optional[list[FlowStateHook[P, R]]] = None,
+        on_cancellation: Optional[list[FlowStateHook[P, R]]] = None,
+        on_crashed: Optional[list[FlowStateHook[P, R]]] = None,
+        on_running: Optional[list[FlowStateHook[P, R]]] = None,
+        job_variables: Optional[dict[str, Any]] = None,
+    ) -> "InfrastructureBoundFlow[P, R]":
+        new_flow = super().with_options(
+            name=name,
+            version=version,
+            retries=retries,
+            retry_delay_seconds=retry_delay_seconds,
+            description=description,
+            flow_run_name=flow_run_name,
+            task_runner=task_runner,
+            timeout_seconds=timeout_seconds,
+            validate_parameters=validate_parameters,
+            persist_result=persist_result,
+            result_storage=result_storage,
+            result_serializer=result_serializer,
+            cache_result_in_memory=cache_result_in_memory,
+            log_prints=log_prints,
+            on_completion=on_completion,
+            on_failure=on_failure,
+            on_cancellation=on_cancellation,
+            on_crashed=on_crashed,
+            on_running=on_running,
+        )
+        new_infrastructure_bound_flow = bind_flow_to_infrastructure(
+            new_flow,
+            self.work_pool,
+            self.worker_cls,
+            job_variables=job_variables
+            if job_variables is not None
+            else self.job_variables,
+        )
+        return new_infrastructure_bound_flow
+
 
 def bind_flow_to_infrastructure(
     flow: Flow[P, R],

prefect_client-3.3.6.dev4/src/prefect/_build_info.py

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.3.6.dev4"
-__build_date__ = "2025-04-24 08:08:45.298549+00:00"
-__git_commit__ = "cd2bf96d18276663c433a99c1e0f1191c100de5f"
-__dirty__ = False

{prefect_client-3.3.6.dev4 → prefect_client-3.3.7.dev1}/src/prefect/_internal/concurrency/threads.py

@@ -174,13 +174,13 @@ class EventLoopThread(Portal):
             # Track the portal running the call
             call.set_runner(self)
 
+            if self._run_once:
+                call.future.add_done_callback(lambda _: self.shutdown())
+
             # Submit the call to the event loop
             assert self._loop is not None
             asyncio.run_coroutine_threadsafe(self._run_call(call), self._loop)
-
             self._submitted_count += 1
-            if self._run_once:
-                call.future.add_done_callback(lambda _: self.shutdown())
 
         return call