isort: llama2c known_third_party

delphi-suite · Mar 3, 2024 · f7cc6b7 · f7cc6b7
1 parent 23f8a55
commit f7cc6b7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -3,4 +3,5 @@ extend-exclude = 'src/llama2c'
 
 [tool.isort]
 profile = 'black'
+known_third_party = ['llama2c']
 extend_skip = ['src/llama2c']
diff --git a/src/delphi/train/training.py b/src/delphi/train/training.py
@@ -10,11 +10,10 @@
 from functools import partial
 
 import torch
-from tqdm import tqdm
-
 from llama2c import Task, model_export
 from llama2c.model import ModelArgs as Llama2ModelArgs
 from llama2c.model import Transformer as Llama2Model
+from tqdm import tqdm
 
 # -----------------------------------------------------------------------------
 # I/O