[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

elk/__main__.py

@@ -64,10 +64,11 @@ def sweep(args):
  # evaluate(args) # args: models_path hidden_states_paths
 
  # for path_model in paths:
- # pass model to eval
- # models_path = ...
- # hidden_states_paths = paths
- # evaluate(args) # args: models_path hidden_states_paths
+ # pass model to eval
+ # models_path = ...
+ # hidden_states_paths = paths
+ # evaluate(args) # args: models_path hidden_states_paths
+
 
 def run():
  parser = ArgumentParser(add_help=False)
@@ -136,7 +137,7 @@ def run():
  # Prevent printing from processes other than the first one
  with redirect_stdout(None) if local_rank != 0 else nullcontext():
  for key in list(vars(args).keys()):
-  print("{}: {}".format(key, vars(args)[key]))
+ print("{}: {}".format(key, vars(args)[key]))
 
  if local_rank != 0:
  logging.getLogger("transformers").setLevel(logging.ERROR)

elk/evaluate/evaluate.py

@@ -18,7 +18,7 @@ def evaluate(args):
 
  ccs_models = torch.load(args.ccs_models_path)
  L = hiddens.shape[1]
- 
+
  statistics = []
  for ccs_model in ccs_models:
  ccs_model.eval()

elk/evaluate/parser.py

@@ -2,11 +2,13 @@
 from pathlib import Path
 from ..files import elk_cache_dir
 
+
 def get_evaluate_parser():
  parser = ArgumentParser(add_help=False)
  add_eval_args(parser)
  return parser
 
+
 def add_eval_args(parser):
  parser.add_argument(
  "ccs_models_path",