Files
onepilot/iqpilot/models/tinygrad_ref.py
2026-03-03 21:39:08 -06:00

37 lines
835 B
Python

import os
from openpilot.common.basedir import BASEDIR
def get_tinygrad_ref():
repo_path = os.path.join(BASEDIR, "tinygrad_repo")
git_path = os.path.join(repo_path, ".git")
try:
if os.path.isdir(git_path):
git_dir = git_path
else:
with open(git_path) as f:
line = f.read().strip()
git_dir = os.path.join(repo_path, line[8:])
with open(os.path.join(git_dir, "HEAD")) as f:
ref = f.read().strip()
if ref.startswith("ref:"):
with open(os.path.join(git_dir, ref.split(" ", 1)[1])) as f:
return f.read().strip()
return ref
except Exception as e:
print(f"Error getting tinygrad_repo ref: {e}")
return None
def main():
current_ref = get_tinygrad_ref()
if current_ref:
print(current_ref)
else:
print("")
if __name__ == "__main__":
main()