fix pyright

EleutherAI · Jul 12, 2023 · f7b96b0 · f7b96b0
1 parent 9988cc5
commit f7b96b0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/elk/training/train.py b/elk/training/train.py
@@ -50,6 +50,7 @@ def create_models_dir(self, out_dir: Path):
  return reporter_dir, lr_dir
 
  def make_eval(self, model, eval_dataset):
+ assert self.out_dir is not None
  return Eval(
  data=replace(
  self.data,