From f7cc6b72a519b8cb8b7426edad0c5b870056df71 Mon Sep 17 00:00:00 2001
From: Jett <jettjaniak@gmail.com>
Date: Sat, 2 Mar 2024 16:24:51 -0800
Subject: [PATCH] isort: llama2c known_third_party

---
 pyproject.toml               | 1 +
 src/delphi/train/training.py | 3 +--
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 25d7eb4e..1d37c3da 100644
--- 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']
\ No newline at end of file
diff --git a/src/delphi/train/training.py b/src/delphi/train/training.py
index 71a9b09e..4ad92951 100644
--- 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