Volume
Reading and Writing Data
How to access the files in storage volumes
How to Write Files in Beam
Your apps run in containers that are read only.
There are two use-cases for saving files: persistent files, that you want to access between tasks, and temporary files that will be deleted when your container spins down.
- Persisting Files: write to a volume.
- Temporary Files: temporary files can be written to the
/tmp
directory in your Beam container, for example you could save an image to/tmp/myimage.png
.
Reading and Writing to Volumes
You can read and write to your Volume like any ordinary Python file:
from beam import function, Volume
VOLUME_PATH = "./model_weights"
@function(
volumes=[Volume(name="model-weights", mount_path=VOLUME_PATH)],
)
def access_files():
# Write files to a volume
with open(f"{VOLUME_PATH}/somefile.txt", "w") as f:
f.write("Writing to the volume!")
# Read files from a volume
with open(f"{VOLUME_PATH}/somefile.txt", "r") as f:
f.read()
Was this page helpful?