MCPcopy Create free account
hub / github.com/FlashSampling/FlashSampling / load_all_data

Function load_all_data

benchmarking/vllm/plot_tpot.py:71–88  ·  view source on GitHub ↗
(results_dir: Path, fmms_name: str)

Source from the content-addressed store, hash-verified

69 variant_dir = model_dir / variant_key
70 if not variant_dir.exists():
71 print(f"Warning: {variant_dir} not found, skipping")
72 return None
73 run_dir = latest_timestamp(variant_dir)
74 return pd.read_csv(run_dir / "summary.csv")
75
76
77def load_all_data(results_dir: Path, fmms_name: str) -> pd.DataFrame:
78 frames = []
79 for model in MODELS:
80 model_dirs = resolve_model_dirs(results_dir, model)
81 if not model_dirs:
82 print(f"Warning: no directory matching {model} in {results_dir}, skipping")
83 continue
84 for model_dir in model_dirs:
85 for variant_key, display_name in [
86 ("baseline", BASELINE_NAME),
87 (FI2_DIR, FI2_NAME),
88 (FMMS_DIR, fmms_name),
89 ]:
90 df = load_variant(model_dir, variant_key)
91 if df is None:

Callers 1

mainFunction · 0.85

Calls 2

resolve_model_dirsFunction · 0.85
load_variantFunction · 0.85

Tested by

no test coverage detected