From 92d3f4bdf96a2d4a5d555b4ddd1b2697f065da09 Mon Sep 17 00:00:00 2001 From: euclidgame Date: Fri, 15 Nov 2024 16:55:31 -0800 Subject: [PATCH] Fix import --- sky/dag.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sky/dag.py b/sky/dag.py index bd1238f132a..40ca5a2c8ef 100644 --- a/sky/dag.py +++ b/sky/dag.py @@ -7,7 +7,7 @@ import networkx as nx -from sky.data.storage import StoreType +from sky.data import storage as storage_lib from sky.utils import common_utils from sky.utils import ux_utils @@ -49,7 +49,7 @@ class TaskEdge: data: Optional[TaskData] = None # TODO(wenjie): Ensure that the region is not None - best_storage: Optional[Tuple[StoreType, Optional[str]]] = None + best_storage: Optional[Tuple[storage_lib.StoreType, Optional[str]]] = None def with_data(self, source_path: str, target_path: str, size_gb: float) -> 'TaskEdge':