Adding docstring to resources decorator

Netflix · Aug 29, 2023 · c5a7817 · c5a7817
1 parent 922b889
commit c5a7817
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/metaflow/plugins/resources_decorator.py b/metaflow/plugins/resources_decorator.py
@@ -25,6 +25,8 @@ class ResourcesDecorator(StepDecorator):
         Number of CPUs required for this step.
     gpu : int, default: 0
         Number of GPUs required for this step.
+    disk : int, default: 10240
+        Disk size (in MB) required for this step. Only applies on Kubernetes.
     memory : int, default: 4096
         Memory size (in MB) required for this step.
     shared_memory : int, optional
@@ -36,7 +38,7 @@ class ResourcesDecorator(StepDecorator):
     defaults = {
         "cpu": "1",
         "gpu": "0",
+        "disk": "10240",
         "memory": "4096",
         "shared_memory": None,
-        "disk": "10240",
     }