Skip to content

Commit 784b6cb

Browse files
committed
Merge branch 'nmb_vlm_models' of https://github.com/VectorInstitute/vector-inference into nmb_vlm_models
2 parents b081625 + c331351 commit 784b6cb

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

tests/vec_inf/cli/test_cli.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -155,9 +155,14 @@ def _mock_truediv(*args):
155155
# Normal case with self and other
156156
self, other = args
157157
specific_paths = {
158-
(str(test_paths["weights_dir"]), "unknown-model"): test_paths["unknown_model"],
158+
(str(test_paths["weights_dir"]), "unknown-model"): test_paths[
159+
"unknown_model"
160+
],
159161
(str(test_paths["log_dir"]), other): test_paths["log_dir"] / other,
160-
(str(test_paths["log_dir"] / "model_family_placeholder"), other): test_paths["log_dir"] / "model_family_placeholder" / other,
162+
(
163+
str(test_paths["log_dir"] / "model_family_placeholder"),
164+
other,
165+
): test_paths["log_dir"] / "model_family_placeholder" / other,
161166
("/home/user", ".vec-inf-logs"): test_paths["log_dir"],
162167
}
163168

0 commit comments

Comments
 (0)