mypy

Author: sanderegg

services/autoscaling/src/simcore_service_autoscaling/modules/cluster_scaling/_utils_computational.py

@@ -3,8 +3,10 @@
 
 from aws_library.ec2 import Resources
 from dask_task_models_library.resource_constraints import (
+    DASK_WORKER_THREAD_RESOURCE_NAME,
     get_ec2_instance_type_from_resources,
 )
+from pydantic import ByteSize
 
 from ...models import DaskTask
 
@@ -17,7 +19,8 @@
 def resources_from_dask_task(task: DaskTask) -> Resources:
     return Resources(
         cpus=task.required_resources.get("CPU", _DEFAULT_MAX_CPU),
-        ram=task.required_resources.get("RAM", _DEFAULT_MAX_RAM),
+        ram=ByteSize(task.required_resources.get("RAM", _DEFAULT_MAX_RAM)),
+        generic_resources={DASK_WORKER_THREAD_RESOURCE_NAME: 1},
     )