add annotation

Signed-off-by: liuxsh9 <liuxiaoshuang4@huawei.com>
ray-project · Mar 3, 2025 · 374ea70 · 374ea70
1 parent b940e26
commit 374ea70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rllib/utils/framework.py b/rllib/utils/framework.py
@@ -54,7 +54,7 @@ def convert_to_tensor(
 def get_device(
     config: "AlgorithmConfig",
     num_gpus_requested: int = 1,
-    custom_resources_requested=None,
+    custom_resources_requested: Optional[dict] = None,
 ):
     """Returns a single device (CPU or some GPU) depending on a config.