type

Signed-off-by: Hao Chen <chenh1024@gmail.com>
ray-project · Dec 21, 2023 · 087f5b4 · 087f5b4
1 parent 893a830
commit 087f5b4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/ray/data/_internal/util.py b/python/ray/data/_internal/util.py
@@ -13,6 +13,7 @@
     TYPE_CHECKING,
     Any,
     Callable,
+    Dict,
     Iterable,
     Iterator,
     List,
@@ -60,7 +61,7 @@
 CLUSTER_RESOURCES_FETCH_INTERVAL_SECONDS = 10
 
 
-def cluster_resources():
+def cluster_resources() -> Dict[str, float]:
     """Fetch Ray cluster resources with cache."""
     global cached_cluster_resources
     global cluster_resources_last_fetch_time