wandb isort fix

delphi-suite · Mar 7, 2024 · 98bfd7d · 98bfd7d
1 parent 5798c87
commit 98bfd7d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -3,7 +3,7 @@ extend-exclude = 'src/llama2c'
 
 [tool.isort]
 profile = 'black'
-known_third_party = ['llama2c']
+known_third_party = ['llama2c', 'wandb']
 extend_skip = ['src/llama2c']
 
 [tool.pytest.ini_options]

diff --git a/src/delphi/train/wandb_utils.py b/src/delphi/train/wandb_utils.py
@@ -1,6 +1,8 @@
 from dataclasses import asdict
 
+import torch
 import wandb
+
 from delphi.train.gigaconfig import GigaConfig
 from delphi.train.utils import EvalData