fix using late-commands and reset-partition-only at the same time

This commit is contained in:
Michael Hudson-Doyle 2023-07-17 13:54:57 +12:00
parent f8bf1105bf
commit b23d553220
2 changed files with 4 additions and 1 deletions

View File

@ -9,3 +9,5 @@ storage:
name: direct
reset-partition: yes
reset-partition-only: yes
late-commands:
- ['true']

View File

@ -129,7 +129,8 @@ class LateController(CmdListController):
def env(self):
env = super().env()
env['TARGET_MOUNT_POINT'] = self.app.base_model.target
if self.app.base_model.target is not None:
env['TARGET_MOUNT_POINT'] = self.app.base_model.target
return env
def start(self):