[pre-commit.ci] pre-commit autoupdate

.pre-commit-config.yaml

@@ -2,7 +2,7 @@
 # See https://pre-commit.com/hooks.html for more hooks
 repos:
 - repo: https://github.com/pre-commit/pre-commit-hooks
- rev: v4.4.0
+ rev: v4.6.0
  hooks:
  - id: trailing-whitespace
  - id: end-of-file-fixer
@@ -11,16 +11,16 @@ repos:
  args: ["--unsafe"]
  - id: check-added-large-files
 - repo: https://github.com/psf/black
- rev: 23.7.0
+ rev: 24.4.2
  hooks:
  - id: black
 - repo: https://github.com/astral-sh/ruff-pre-commit
- rev: 'v0.0.278'
+ rev: 'v0.5.2'
  hooks:
  - id: ruff
  args: [--fix, --exit-non-zero-on-fix]
 - repo: https://github.com/codespell-project/codespell
- rev: v2.2.5
+ rev: v2.3.0
  hooks:
  - id: codespell
  # The promptsource templates spuriously get flagged without this

elk/extraction/extraction.py

@@ -1,4 +1,5 @@
 """Functions for extracting the hidden states of a model."""
+
 import os
 from collections import defaultdict
 from dataclasses import InitVar, dataclass, replace

elk/plotting/visualize.py

@@ -78,9 +78,9 @@ def render(
  y=dataset_data["auroc_estimate"],
  mode="lines",
  name=ensemble,
- showlegend=False
- if dataset_name != unique_datasets[0]
- else True,
+ showlegend=(
+  False if dataset_name != unique_datasets[0] else True
+ ),
  line=dict(color=color_map[ensemble]),
  ),
  row=row,

elk/promptsource/templates.py

@@ -215,9 +215,11 @@ def _escape_pipe(cls, example):
  # Replaces any occurrences of the "|||" separator in the example, which
  # which will be replaced back after splitting
  protected_example = {
- key: value.replace("|||", cls.pipe_protector)
- if isinstance(value, str)
- else value
+ key: (
+ value.replace("|||", cls.pipe_protector)
+ if isinstance(value, str)
+ else value
+ )
  for key, value in example.items()
  }
  return protected_example

elk/training/platt_scaling.py

@@ -12,8 +12,7 @@ class PlattMixin(ABC):
  scale: nn.Parameter
 
  @abstractmethod
- def __call__(self, *args: Any, **kwds: Any) -> Any:
- ...
+ def __call__(self, *args: Any, **kwds: Any) -> Any: ...
 
  def platt_scale(self, labels: Tensor, hiddens: Tensor, max_iter: int = 100):
  """Fit the scale and bias terms to data with LBFGS.