available_memory step: use bytes as output type

arcalot · Dec 5, 2024 · 2a494ca · 2a494ca
1 parent ed06a83
commit 2a494ca
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/arcaflow_plugin_utilities/arcaflow_plugin_utilities.py b/arcaflow_plugin_utilities/arcaflow_plugin_utilities.py
@@ -65,6 +65,7 @@ class SuccessOutputNumCpu:
 class SuccessOutputAvailableMemory:
     available_memory: typing.Annotated[
         int,
+        schema.units(schema.UNIT_BYTE),
         schema.name("available memory"),
         schema.description("Amount of available memory (in bytes)"),
     ]