Jl harness (#136)

- adds logging to output.py in harness utils, to remove the need to
write the logging function for every harness
- writes a more convenient function to call keras and parse keras args
within misc.py in harness_utils, does not remove any functionality from
other harnesses
- added function to call time function while rounding to int 
- added default paus value for press n times
- dota2: changing to path instead of os
- ac shadows: os to path conversion, uses imported functions instead of
writing
- general linting 
- the last of us part 1 lint
- new the last of us part 2 harness
This commit is contained in:
j-lin-lmg
2025-05-13 14:53:25 -07:00
committed by GitHub
parent 1e2abd47b3
commit 6f47569db6
7 changed files with 405 additions and 91 deletions

View File

@@ -1,13 +1,13 @@
"""Dota 2 test script"""
import logging
import os
from pathlib import Path
import time
import pyautogui as gui
import pydirectinput as user
import sys
from dota2_utils import get_resolution, copy_replay, copy_config, get_args
sys.path.insert(1, os.path.join(sys.path[0], '..'))
sys.path.insert(1, str(Path(sys.path[0]).parent))
from harness_utils.output import (
setup_log_directory,
@@ -22,8 +22,8 @@ from harness_utils.steam import exec_steam_game
from harness_utils.artifacts import ArtifactManager, ArtifactType
SCRIPT_DIRECTORY = os.path.dirname(os.path.realpath(__file__))
LOG_DIRECTORY = os.path.join(SCRIPT_DIRECTORY, "run")
SCRIPT_DIRECTORY = Path(__file__).resolve().parent
LOG_DIRECTORY = SCRIPT_DIRECTORY / "run"
PROCESS_NAME = "dota2.exe"
STEAM_GAME_ID = 570
@@ -198,9 +198,9 @@ def run_benchmark():
try:
start_time, end_time = run_benchmark()
height, width = get_resolution()
res_height, res_width = get_resolution()
report = {
"resolution": format_resolution(width, height),
"resolution": format_resolution(res_width, res_height),
"start_time": seconds_to_milliseconds(start_time),
"end_time": seconds_to_milliseconds(end_time)
}