0.0.0+develop
flytekitplugins.memray
Directory
Classes
Class | Description |
---|---|
memray_profiling |
Abstract class for class decorators. |
flytekitplugins.memray.memray_profiling
Abstract class for class decorators. We can attach config on the decorator class and use it in the upper level.
class memray_profiling(
task_function: typing.Optional[typing.Callable],
native_traces: bool,
trace_python_allocators: bool,
follow_fork: bool,
memory_interval_ms: int,
memray_html_reporter: str,
memray_reporter_args: typing.Optional[typing.List[str]],
)
Memray profiling plugin.
Parameter | Type |
---|---|
task_function |
typing.Optional[typing.Callable] |
native_traces |
bool |
trace_python_allocators |
bool |
follow_fork |
bool |
memory_interval_ms |
int |
memray_html_reporter |
str |
memray_reporter_args |
typing.Optional[typing.List[str]] |
Methods
Method | Description |
---|---|
execute() |
This method will be called when the decorated function is called. |
generate_flytedeck_html() |
|
get_extra_config() |
Get the config of the decorator. |
execute()
def execute(
args,
kwargs,
)
This method will be called when the decorated function is called.
Parameter | Type |
---|---|
args |
*args |
kwargs |
**kwargs |
generate_flytedeck_html()
def generate_flytedeck_html(
reporter,
bin_filepath,
)
Parameter | Type |
---|---|
reporter |
|
bin_filepath |
get_extra_config()
def get_extra_config()
Get the config of the decorator.