updated generation process

1 files changed, 3 insertions, 3 deletions

diff --git a/src/gap_pipeline/cli.py b/src/gap_pipeline/cli.py
index 105cc97..e79e3e1 100644
--- a/src/gap_pipeline/cli.py
+++ b/src/gap_pipeline/cli.py
@@ -16,7 +16,7 @@ from .offline import (
     summarize_run,
     validate_public_dataset,
 )
-from .pipeline import KernelPipeline, PipelineConfig
+from .paper_pipeline import PaperKernelPipeline, PaperPipelineConfig
 from .release import export_release
 from .store import RunStore
 from .surface import SurfacePipeline
@@ -36,7 +36,7 @@ async def generate_kernel(args: argparse.Namespace) -> None:
     if args.item_id not in records:
         raise SystemExit(f"item ID {args.item_id!r} not found in {args.dataset}")
     item = CanonicalItem.from_public_record(records[args.item_id])
-    config = PipelineConfig(
+    config = PaperPipelineConfig(
         proposer_model=args.proposer_model,
         judge_model=args.judge_model,
     )
@@ -48,7 +48,7 @@ async def generate_kernel(args: argparse.Namespace) -> None:
         )
         for judge_id in range(1, 6)
     ]
-    pipeline = KernelPipeline(
+    pipeline = PaperKernelPipeline(
         proposer=proposer,
         judges=judges,
         store=RunStore(args.run_dir, item.item_id),