MMLU-by-task-Leaderboard / save_for_regression.py
Corey Morris
changed to save and load in a directory
dd61816
raw
history blame
1.86 kB
# when run
# checks if there is uncommitted code
# if there is uncommitted code, ti retuns an error
# if there is no uncommitted code, it saves the dataframe as a parquet file with the commit hash in the name
import pytest
import pandas as pd
from result_data_processor import ResultDataProcessor
import os
import subprocess
def check_git_changes(repo_path):
try:
# Change to the repository directory
original_path = os.getcwd()
os.chdir(repo_path)
# Run the git status command
result = subprocess.run(['git', 'status', '--porcelain'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
# Check the result
if result.returncode != 0:
print(f"Error checking git status: {result.stderr}")
return False, False
# Check for tracked and untracked changes
tracked_changes = any(line[:2].strip() != '??' for line in result.stdout.splitlines())
untracked_changes = any(line[:2] == '??' for line in result.stdout.splitlines())
return tracked_changes, untracked_changes
finally:
# Change back to the original directory
os.chdir(original_path)
if __name__ == '__main__':
tracked_changes, untracked_changes = check_git_changes('.')
if tracked_changes:
print("There are tracked changes")
else:
print("There are no tracked changes")
df_current = ResultDataProcessor().data
last_commit = os.popen('git rev-parse HEAD').read().strip()
print(last_commit)
# save the current output to a file
df_current.to_parquet(f'dataframe_history/output_{last_commit}.parquet', index=True)
print("Saved output to file")
if untracked_changes:
print("There are untracked changes")
else:
print("There are no untracked changes")