diff --git a/.readthedocs.yaml b/.readthedocs.yaml new file mode 100644 index 000000000..e2645f949 --- /dev/null +++ b/.readthedocs.yaml @@ -0,0 +1,20 @@ +# .readthedocs.yaml +# Read the Docs configuration file +# See https://docs.readthedocs.io/en/stable/config-file/v2.html for details + +# Required +version: 2 + +# Set the version of Python and other tools you might need +build: + os: ubuntu-22.04 + tools: + python: "3.11" + +mkdocs: + configuration: mkdocs.yml + +# Optionally declare the Python requirements required to build your docs +python: + install: + - requirements: docs/requirements.txt diff --git a/pilot/utils.py b/pilot/utils.py index 0179d12c2..607b83251 100644 --- a/pilot/utils.py +++ b/pilot/utils.py @@ -33,7 +33,7 @@ def get_gpu_memory(max_gpus=None): allocated_memory = torch.cuda.memory_allocated() / (1024 ** 3) available_memory = total_memory - allocated_memory gpu_memory.append(available_memory) - return gpu_memory + return gpu_memory