Memray Profiling
flytekitplugins-memray
FlytekitDeveloper Toolsmemraymemoryprofiling
Memray tracks and reports memory allocations, both in python code and in compiled extension modules.
Install
pip install flytekitplugins-memrayQuick Start(example, may need adjustment)
See full examplespip install flytekitplugins-memray
from flytekit import task, workflow
from flytekitplugins.memray import memray_profiling
@task
def my_task() -> None:
memray_profiling(...)
@workflow
def my_workflow() -> None:
my_task()Available Imports (1)
taskmemray_profiling
Task for Memray Profiling.
from flytekitplugins.memray import memray_profiling
Dependencies
memray
Related Plugins
Anthropic Claude
This plugin provides integration between Flyte and Anthropic's Claude API, enabling you to use Flyte tasks as tools for Claude agents.
Deck
This plugin provides additional renderers to improve task visibility within Flytekit.
Envd
envd is a command-line tool that helps you create the container-based development environment for AI/ML.
Flyte Interactive
FlyteInteractive plugin provides users' favorite interface to develop and debug a flyte task interactively. We support vscode, jupyter (WIP), and neovim (WIP).
Package Info
Min Flytekit1.12.0
Modules1
Downloads
Last day13
Last week53
Last month316
