From 1e278887ef3439a568cfb9eaeb4590111261fb36 Mon Sep 17 00:00:00 2001
From: Yu Li <lyo.gavin@gmail.com>
Date: Tue, 19 Dec 2023 11:57:45 -0600
Subject: [PATCH] fix import

---
 air_llm/tests/test_automodel.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/air_llm/tests/test_automodel.py b/air_llm/tests/test_automodel.py
index 8224295..493af23 100644
--- a/air_llm/tests/test_automodel.py
+++ b/air_llm/tests/test_automodel.py
@@ -2,9 +2,9 @@ import sys
 import unittest
 
 import torch
-sys.path.insert(0, '../airllm')
+#sys.path.insert(0, '../airllm')
 
-from auto_model import AutoModel
+from airllm.auto_model import AutoModel