Skip to content

Commit c56708e

Browse files
fix mount path resolution
1 parent c7688fe commit c56708e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

keras_remote/runner/remote_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def _resolve(obj):
154154
if isinstance(obj, dict) and obj.get("__data_ref__"):
155155
# Volume-mounted data refs are handled by Kubernetes, skip download
156156
if obj.get("mount_path") is not None:
157-
return obj
157+
return obj["mount_path"]
158158
local_dir = os.path.join(DATA_DIR, str(counter))
159159
counter += 1
160160
_download_data(obj, local_dir, storage_client)

0 commit comments

Comments
 (0)