diff --git a/tests/models/model_test.py b/tests/models/model_test.py index 7a8626b4b..67f14498a 100644 --- a/tests/models/model_test.py +++ b/tests/models/model_test.py @@ -628,7 +628,7 @@ def load_tokenizer(self, model_id_or_path, trust_remote_code=False): @classmethod def load_dataset(cls, tokenizer=None, rows: int = 0): try: - dataset = load_dataset(path="/monster/data/_ci_/nm-calibration", name="LLM", split="train") + dataset = load_dataset(path="/monster/data/model/dataset/nm-calibration", name="LLM", split="train") except Exception as exc: # pragma: no cover - exercised in fallbacks log.warning("load_dataset failed; falling back to local parquet: %s", exc) dataset = cls._load_calibration_parquet() @@ -639,7 +639,7 @@ def load_dataset(cls, tokenizer=None, rows: int = 0): @staticmethod def _load_calibration_parquet(): - parquet_path = Path("/monster/data/_ci_/nm-calibration/llm.parquet").expanduser() + parquet_path = Path("/monster/data/model/dataset/nm-calibration/llm.parquet").expanduser() if not parquet_path.exists(): raise FileNotFoundError(f"Calibration parquet not found at {parquet_path}")