Skip to content

Commit 3e186cf

Browse files
committed
chore: apply pre-commit formatting
1 parent 6e9535d commit 3e186cf

File tree

4 files changed

+11
-33
lines changed

4 files changed

+11
-33
lines changed

dashboard/pages/1_comm.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,7 @@
2222
)
2323

2424
# Page configuration
25-
st.set_page_config(
26-
page_title="通信测试分析 | InfiniMetrics", page_icon="🔗", layout="wide"
27-
)
25+
st.set_page_config(page_title="通信测试分析 | InfiniMetrics", page_icon="🔗", layout="wide")
2826

2927
# Initialize session state
3028
if "data_loader" not in st.session_state:

dashboard/pages/2_ops.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,9 +16,7 @@
1616
plot_timeseries_auto,
1717
)
1818

19-
st.set_page_config(
20-
page_title="算子测试分析 | InfiniMetrics", page_icon="⚡", layout="wide"
21-
)
19+
st.set_page_config(page_title="算子测试分析 | InfiniMetrics", page_icon="⚡", layout="wide")
2220

2321
if "data_loader" not in st.session_state:
2422
st.session_state.data_loader = InfiniMetricsDataLoader()

dashboard/pages/3_infer.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,9 +16,7 @@
1616
create_summary_table_infer,
1717
)
1818

19-
st.set_page_config(
20-
page_title="推理测试分析 | InfiniMetrics", page_icon="🤖", layout="wide"
21-
)
19+
st.set_page_config(page_title="推理测试分析 | InfiniMetrics", page_icon="🤖", layout="wide")
2220

2321
if "data_loader" not in st.session_state:
2422
st.session_state.data_loader = InfiniMetricsDataLoader()

dashboard/utils/visualizations.py

Lines changed: 8 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -230,21 +230,15 @@ def create_summary_table(test_result: Dict[str, Any]) -> pd.DataFrame:
230230
accelerators = machine.get("accelerators", [])
231231
if accelerators:
232232
acc = accelerators[0]
233-
summary_data.append(
234-
{"指标": "GPU型号", "数值": acc.get("model", "Unknown")}
235-
)
236-
summary_data.append(
237-
{"指标": "GPU数量", "数值": acc.get("count", "Unknown")}
238-
)
233+
summary_data.append({"指标": "GPU型号", "数值": acc.get("model", "Unknown")})
234+
summary_data.append({"指标": "GPU数量", "数值": acc.get("count", "Unknown")})
239235
summary_data.append(
240236
{
241237
"指标": "显存/卡",
242238
"数值": f"{acc.get('memory_gb_per_card', 'Unknown')} GB",
243239
}
244240
)
245-
summary_data.append(
246-
{"指标": "CUDA版本", "数值": acc.get("cuda", "Unknown")}
247-
)
241+
summary_data.append({"指标": "CUDA版本", "数值": acc.get("cuda", "Unknown")})
248242

249243
# Test config summary
250244
config = test_result.get("config", {})
@@ -387,21 +381,15 @@ def create_summary_table_infer(test_result: dict) -> pd.DataFrame:
387381
{"指标": "模型", "数值": cfg.get("model", "")},
388382
{
389383
"指标": "batch",
390-
"数值": (cfg.get("infer_args", {}) or {}).get(
391-
"static_batch_size", "unknown"
392-
),
384+
"数值": (cfg.get("infer_args", {}) or {}).get("static_batch_size", "unknown"),
393385
},
394386
{
395387
"指标": "prompt_tok",
396-
"数值": (cfg.get("infer_args", {}) or {}).get(
397-
"prompt_token_num", "unknown"
398-
),
388+
"数值": (cfg.get("infer_args", {}) or {}).get("prompt_token_num", "unknown"),
399389
},
400390
{
401391
"指标": "output_tok",
402-
"数值": (cfg.get("infer_args", {}) or {}).get(
403-
"output_token_num", "unknown"
404-
),
392+
"数值": (cfg.get("infer_args", {}) or {}).get("output_token_num", "unknown"),
405393
},
406394
{"指标": "warmup", "数值": cfg.get("warmup_iterations", "unknown")},
407395
{"指标": "measured", "数值": cfg.get("measured_iterations", "unknown")},
@@ -423,9 +411,7 @@ def create_summary_table_ops(test_result: dict) -> pd.DataFrame:
423411

424412
rows.append({"指标": "testcase", "数值": test_result.get("testcase", "")})
425413
# Try to get operator name from config
426-
rows.append(
427-
{"指标": "算子", "数值": cfg.get("operator", cfg.get("op_name", "Unknown"))}
428-
)
414+
rows.append({"指标": "算子", "数值": cfg.get("operator", cfg.get("op_name", "Unknown"))})
429415

430416
# Environment info
431417
env = test_result.get("environment", {})
@@ -441,9 +427,7 @@ def create_summary_table_ops(test_result: dict) -> pd.DataFrame:
441427
# Scalar metrics summary
442428
scalars = [m for m in test_result.get("metrics", []) if m.get("type") == "scalar"]
443429
for m in scalars:
444-
rows.append(
445-
{"指标": m.get("name"), "数值": f"{m.get('value')} {m.get('unit','')}"}
446-
)
430+
rows.append({"指标": m.get("name"), "数值": f"{m.get('value')} {m.get('unit','')}"})
447431

448432
# Common config fields fallback
449433
for k in [

0 commit comments

Comments
 (0)