summaryrefslogtreecommitdiff
path: root/kvmd/plugins/msd/relay/drive.py
diff options
context:
space:
mode:
Diffstat (limited to 'kvmd/plugins/msd/relay/drive.py')
-rw-r--r--kvmd/plugins/msd/relay/drive.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/kvmd/plugins/msd/relay/drive.py b/kvmd/plugins/msd/relay/drive.py
index 2e95e837..e1301f99 100644
--- a/kvmd/plugins/msd/relay/drive.py
+++ b/kvmd/plugins/msd/relay/drive.py
@@ -32,7 +32,7 @@ from typing import Optional
from .... import aiotools
from .... import aiofs
-from .. import MsdImageWriter
+from .. import MsdFileWriter
# =====
@@ -121,7 +121,7 @@ class DeviceInfo:
image=image_info,
)
- async def write_image_info(self, device_writer: MsdImageWriter, complete: bool) -> bool:
+ async def write_image_info(self, device_writer: MsdFileWriter, complete: bool) -> bool:
device_file = device_writer.get_file()
state = device_writer.get_state()
image_info = ImageInfo(state["name"], state["written"], complete)