Fixing imports.

Author: Vitaliy Zakaznikov

testflows/github/hetzner/runners/metrics.py

@@ -17,7 +17,8 @@
 import logging
 from datetime import datetime
 
-from github.WorkflowRun import WorkflowRun, WorkflowRunJob
+from github.WorkflowRun import WorkflowRun
+from github.WorkflowJob import WorkflowJob
 from prometheus_client import Counter, Gauge, Histogram, Info
 from .estimate import get_server_price
 from .constants import standby_server_name_prefix, recycle_server_name_prefix
@@ -657,7 +658,7 @@ def update_runners(
         ).set(count)
 
 
-def update_jobs(run_jobs: list[(WorkflowRun, WorkflowRunJob)]):
+def update_jobs(run_jobs: list[(WorkflowRun, WorkflowJob)]):
     """Update all job-related metrics."""
     queued_count = 0
     running_count = 0

testflows/github/hetzner/runners/scale_up.py

@@ -370,7 +370,7 @@ def job_matches_labels(job_labels, with_label):
     return True
 
 
-def filtered_run_jobs(workflow_runs, with_label):
+def filtered_run_jobs(workflow_runs: list[WorkflowRun], with_label: list[str]):
     """Filter jobs to select only queued or in progress and match with_label criteria."""
     run_jobs = []
     for run in workflow_runs: