From 0f683f639cfb34ce2c3995f904e063e3fbb56cbb Mon Sep 17 00:00:00 2001
From: Benjamin Ye <benjamin.ye@georgian.io>
Date: Tue, 9 Apr 2024 00:49:54 -0400
Subject: [PATCH] fix lint

---
 llmtune/pydantic_models/config_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/llmtune/pydantic_models/config_model.py b/llmtune/pydantic_models/config_model.py
index 1095656..e0f9976 100644
--- a/llmtune/pydantic_models/config_model.py
+++ b/llmtune/pydantic_models/config_model.py
@@ -1,7 +1,7 @@
 from typing import List, Literal, Optional, Union
 
 import torch
-from pydantic import BaseModel, ConfigDict, Field, FilePath, validator
+from pydantic import BaseModel, Field, FilePath, validator
 
 
 # TODO: Refactor this into multiple files...