Differential D8684 Diff 40955 examples/docs_snippets/docs_snippets/concepts/io_management/io_manager_per_output.py
Changeset View
Changeset View
Standalone View
Standalone View
examples/docs_snippets/docs_snippets/concepts/io_management/io_manager_per_output.py
# start_marker | # start_marker | ||||
from dagster import ModeDefinition, OutputDefinition, fs_io_manager, mem_io_manager, pipeline, solid | from dagster import ( | ||||
ModeDefinition, | |||||
OutputDefinition, | |||||
fs_io_manager, | |||||
mem_io_manager, | |||||
pipeline, | |||||
solid, | |||||
) | |||||
@solid(output_defs=[OutputDefinition(io_manager_key="fs")]) | @solid(output_defs=[OutputDefinition(io_manager_key="fs")]) | ||||
def solid1(): | def solid1(): | ||||
return 1 | return 1 | ||||
@solid(output_defs=[OutputDefinition(io_manager_key="mem")]) | @solid(output_defs=[OutputDefinition(io_manager_key="mem")]) | ||||
def solid2(a): | def solid2(a): | ||||
return a + 1 | return a + 1 | ||||
@pipeline(mode_defs=[ModeDefinition(resource_defs={"fs": fs_io_manager, "mem": mem_io_manager})]) | @pipeline( | ||||
mode_defs=[ | |||||
ModeDefinition( | |||||
resource_defs={"fs": fs_io_manager, "mem": mem_io_manager} | |||||
) | |||||
] | |||||
) | |||||
def my_pipeline(): | def my_pipeline(): | ||||
solid2(solid1()) | solid2(solid1()) | ||||
# end_marker | # end_marker |