add typing

EleutherAI · Jul 28, 2023 · 94a498e · 94a498e
1 parent e8872c8
commit 94a498e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/elk/training/ccs_reporter.py b/elk/training/ccs_reporter.py
@@ -100,7 +100,7 @@ def __init__(
 
  self.norm = None
 
- layers = [
+ layers: list[nn.Module] = [
  nn.Linear(
  in_features,
  1 if cfg.num_layers < 2 else hidden_size,