flytekitplugins-sleep 1.16.27__tar.gz

flytekitplugins_sleep-1.16.27/PKG-INFO

@@ -0,0 +1,27 @@
+Metadata-Version: 2.4
+Name: flytekitplugins-sleep
+Version: 1.16.27
+Summary: This package holds the core-sleep plugin for flytekit
+Author: flyteorg
+Author-email: admin@flyte.org
+License: apache2
+Classifier: Intended Audience :: Science/Research
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development
+Classifier: Topic :: Software Development :: Libraries
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.9
+Requires-Dist: flytekit>=1.15.1
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: license
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

flytekitplugins_sleep-1.16.27/flytekitplugins/sleep/__init__.py

@@ -0,0 +1,11 @@
+"""
+.. currentmodule:: flytekitplugins.sleep
+
+.. autosummary::
+   :template: custom.rst
+   :toctree: generated/
+
+   Sleep
+"""
+
+from .task import Sleep

flytekitplugins_sleep-1.16.27/flytekitplugins/sleep/task.py

@@ -0,0 +1,46 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any, Callable, Dict, Optional
+
+from flytekit.configuration import SerializationSettings
+from flytekit.core.python_function_task import PythonFunctionTask
+from flytekit.core.task import TaskPlugins
+
+
+@dataclass
+class Sleep:
+    """
+    Route a task to the backend ``core-sleep`` plugin.
+
+    The sleep duration is provided as a normal task input, not plugin config.
+    No container is launched; the backend handles the sleep directly.
+
+    Usage::
+
+        from flytekitplugins.sleep import Sleep
+        from flytekit import task
+        from datetime import timedelta
+
+        @task(task_config=Sleep())
+        def sleep_for(duration: timedelta) -> None:
+            pass  # only runs locally; backend executes the sleep
+    """
+
+
+class SleepFunctionTask(PythonFunctionTask[Sleep]):
+    _TASK_TYPE = "core-sleep"
+
+    def __init__(self, task_config: Optional[Sleep], task_function: Callable, **kwargs):
+        super().__init__(
+            task_config=task_config or Sleep(),
+            task_function=task_function,
+            task_type=self._TASK_TYPE,
+            **kwargs,
+        )
+
+    def get_custom(self, settings: SerializationSettings) -> Dict[str, Any]:
+        return {}
+
+
+TaskPlugins.register_pythontask_plugin(Sleep, SleepFunctionTask)

flytekitplugins_sleep-1.16.27/flytekitplugins_sleep.egg-info/PKG-INFO

@@ -0,0 +1,27 @@
+Metadata-Version: 2.4
+Name: flytekitplugins-sleep
+Version: 1.16.27
+Summary: This package holds the core-sleep plugin for flytekit
+Author: flyteorg
+Author-email: admin@flyte.org
+License: apache2
+Classifier: Intended Audience :: Science/Research
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development
+Classifier: Topic :: Software Development :: Libraries
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.9
+Requires-Dist: flytekit>=1.15.1
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: license
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

flytekitplugins_sleep-1.16.27/flytekitplugins_sleep.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+setup.py
+flytekitplugins/sleep/__init__.py
+flytekitplugins/sleep/task.py
+flytekitplugins_sleep.egg-info/PKG-INFO
+flytekitplugins_sleep.egg-info/SOURCES.txt
+flytekitplugins_sleep.egg-info/dependency_links.txt
+flytekitplugins_sleep.egg-info/entry_points.txt
+flytekitplugins_sleep.egg-info/namespace_packages.txt
+flytekitplugins_sleep.egg-info/requires.txt
+flytekitplugins_sleep.egg-info/top_level.txt
+tests/test_sleep.py

flytekitplugins_sleep-1.16.27/flytekitplugins_sleep.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

flytekitplugins_sleep-1.16.27/flytekitplugins_sleep.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[flytekit.plugins]
+sleep = flytekitplugins.sleep

flytekitplugins_sleep-1.16.27/flytekitplugins_sleep.egg-info/namespace_packages.txt

@@ -0,0 +1 @@
+flytekitplugins

flytekitplugins_sleep-1.16.27/flytekitplugins_sleep.egg-info/requires.txt

@@ -0,0 +1 @@
+flytekit>=1.15.1

flytekitplugins_sleep-1.16.27/flytekitplugins_sleep.egg-info/top_level.txt

@@ -0,0 +1 @@
+flytekitplugins

flytekitplugins_sleep-1.16.27/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

flytekitplugins_sleep-1.16.27/setup.py

@@ -0,0 +1,36 @@
+from setuptools import setup
+
+PLUGIN_NAME = "sleep"
+
+microlib_name = f"flytekitplugins-{PLUGIN_NAME}"
+
+__version__ = "1.16.27"
+
+setup(
+    title="Sleep",
+    title_expanded="Flytekit Sleep Plugin",
+    name=microlib_name,
+    version=__version__,
+    author="flyteorg",
+    author_email="admin@flyte.org",
+    description="This package holds the core-sleep plugin for flytekit",
+    namespace_packages=["flytekitplugins"],
+    packages=[f"flytekitplugins.{PLUGIN_NAME}"],
+    install_requires=["flytekit>=1.15.1"],
+    license="apache2",
+    python_requires=">=3.9",
+    classifiers=[
+        "Intended Audience :: Science/Research",
+        "Intended Audience :: Developers",
+        "License :: OSI Approved :: Apache Software License",
+        "Programming Language :: Python :: 3.9",
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
+        "Topic :: Scientific/Engineering",
+        "Topic :: Scientific/Engineering :: Artificial Intelligence",
+        "Topic :: Software Development",
+        "Topic :: Software Development :: Libraries",
+        "Topic :: Software Development :: Libraries :: Python Modules",
+    ],
+    entry_points={"flytekit.plugins": [f"{PLUGIN_NAME}=flytekitplugins.{PLUGIN_NAME}"]},
+)

flytekitplugins_sleep-1.16.27/tests/test_sleep.py

@@ -0,0 +1,69 @@
+from collections import OrderedDict
+from datetime import timedelta
+
+from flytekitplugins.sleep import Sleep
+from flytekitplugins.sleep.task import SleepFunctionTask
+
+from flytekit import task, workflow
+from flytekit.configuration import Image, ImageConfig, SerializationSettings
+from flytekit.extend import get_serializable
+
+default_img = Image(name="default", fqn="test", tag="tag")
+serialization_settings = SerializationSettings(
+    project="proj",
+    domain="dom",
+    version="123",
+    image_config=ImageConfig(default_image=default_img, images=[default_img]),
+    env={},
+)
+
+
+def test_task_type():
+    @task(task_config=Sleep())
+    def sleep_for(duration: timedelta) -> None:
+        pass
+
+    assert isinstance(sleep_for, SleepFunctionTask)
+    assert sleep_for.task_type == "core-sleep"
+
+
+def test_serialization():
+    @task(task_config=Sleep())
+    def sleep_for(duration: timedelta) -> None:
+        pass
+
+    task_spec = get_serializable(OrderedDict(), serialization_settings, sleep_for)
+
+    assert task_spec.template.type == "core-sleep"
+    assert task_spec.template.custom == {}
+
+    inputs = task_spec.template.interface.inputs
+    assert "duration" in inputs
+    assert len(inputs) == 1
+
+    outputs = task_spec.template.interface.outputs
+    assert len(outputs) == 0
+
+
+def test_local_execution_calls_function():
+    called = []
+
+    @task(task_config=Sleep())
+    def sleep_for(duration: timedelta) -> None:
+        called.append(duration)
+
+    sleep_for(duration=timedelta(seconds=1))
+    assert called == [timedelta(seconds=1)]
+
+
+def test_workflow_integration():
+    @task(task_config=Sleep())
+    def sleep_for(duration: timedelta) -> None:
+        pass
+
+    @workflow
+    def wf(duration: timedelta) -> None:
+        sleep_for(duration=duration)
+
+    spec = get_serializable(OrderedDict(), serialization_settings, wf)
+    assert len(spec.template.nodes) == 1