minor fix

lean-dojo · Sep 18, 2024 · cef522f · cef522f
1 parent c1efbf6
commit cef522f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/lean_dojo/data_extraction/lean.py b/src/lean_dojo/data_extraction/lean.py
@@ -434,7 +434,11 @@ def get_lean4_version_from_config(toolchain: str) -> str:
     """Return the required Lean version given a ``lean-toolchain`` config."""
     m = _LEAN4_VERSION_REGEX.fullmatch(toolchain.strip())
     assert m is not None, "Invalid config."
-    return m["version"]
+    v = m["version"]
+    if not v.startswith("v"):
+        assert v[0].isnumeric()
+        v = "v" + v
+    return v
 
 
 def get_lean4_commit_from_config(config_dict: Dict[str, Any]) -> str: