fix

scripts/benchmarks/__init__.py

@@ -1 +1,5 @@
 """Benchmark helpers."""
+
+from .deepseek_v3_moe import DTYPE_MAP, benchmark_deepseek_v3
+
+__all__ = ["benchmark_deepseek_v3", "DTYPE_MAP"]

scripts/benchmarks/deepseek_v3_moe_sweep.py

@@ -6,9 +6,15 @@ from __future__ import annotations
 
 import argparse
 import csv
+import sys
 from pathlib import Path
 from types import SimpleNamespace
 
+CURRENT_DIR = Path(__file__).resolve().parent
+PACKAGE_ROOT = CURRENT_DIR.parent
+if str(PACKAGE_ROOT) not in sys.path:
+    sys.path.insert(0, str(PACKAGE_ROOT))
+
 from scripts.benchmarks.deepseek_v3_moe import (
     DTYPE_MAP,
     benchmark_deepseek_v3,