fix remnant of renaming

elk/training/sweep.py

@@ -5,7 +5,6 @@
 from datasets import get_dataset_config_info
 from transformers import AutoConfig
 
-from ..evaluation import Eval
 from ..extraction import Extract
 from ..files import memorably_named_dir, sweeps_dir
 from ..plotting.visualize import visualize_sweep

elk/training/train.py

@@ -51,8 +51,8 @@ def create_models_dir(self, out_dir: Path):
 
  def make_eval(self, model, eval_dataset):
  return Eval(
- extract=replace(
- self.extract,
+ data=replace(
+ self.data,
  model=model,
  datasets=(eval_dataset,),
  ),