Skip to content

Commit 01ce209

Browse files
authored
fixed incorrect docker image name (#49)
1 parent 9e1a163 commit 01ce209

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

modelconverter/utils/docker_utils.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -108,15 +108,12 @@ def docker_build(
108108
def get_docker_image(
109109
target: Literal["rvc2", "rvc3", "rvc4", "hailo"],
110110
bare_tag: str,
111-
version: Optional[str] = None,
111+
version: str,
112112
) -> str:
113113
check_docker()
114114

115115
client = docker.from_env()
116-
if version is not None:
117-
tag = f"{version}-{bare_tag}"
118-
else:
119-
tag = bare_tag
116+
tag = f"{version}-{bare_tag}"
120117

121118
image = f"luxonis/modelconverter-{target}:{tag}"
122119

@@ -147,6 +144,7 @@ def docker_exec(
147144
use_gpu: bool,
148145
version: Optional[str] = None,
149146
) -> None:
147+
version = version or get_default_target_version(target)
150148
image = get_docker_image(target, bare_tag, version)
151149

152150
with tempfile.NamedTemporaryFile(delete=False) as f:

0 commit comments

Comments
 (0)