Skip to content

Commit 3aff39f

Browse files
committed
fix: regex for model name add dot
1 parent ba2d522 commit 3aff39f

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/balatrollm/data_collection.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,7 @@ def __init__(self, config: Config, base_dir: Path, run_dir: Path | None = None):
129129
self._request_count = 0
130130

131131
def _generate_run_dir(self, base_dir: Path) -> Path:
132-
assert re.match(r"^[a-z0-9-]+/[a-z0-9:-]+$", self.config.model), (
132+
assert re.match(r"^[a-z0-9.-]+/[a-z0-9:.-]+$", self.config.model), (
133133
f"Invalid vendor/model format: {self.config.model}"
134134
)
135135
vendor, model = self.config.model.split("/", 1)

0 commit comments

Comments
 (0)