From 7c13c0e402aaa0efc64e3c4b720d523a05bdd9cf Mon Sep 17 00:00:00 2001
From: Yu Li <lyo.gavin@gmail.com>
Date: Tue, 19 Dec 2023 13:55:57 -0600
Subject: [PATCH] fix test import

---
 air_llm/__init__.py             | 0
 air_llm/tests/__init__.py       | 0
 air_llm/tests/test_automodel.py | 5 ++---
 3 files changed, 2 insertions(+), 3 deletions(-)
 create mode 100644 air_llm/__init__.py
 create mode 100644 air_llm/tests/__init__.py

diff --git a/air_llm/__init__.py b/air_llm/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/air_llm/tests/__init__.py b/air_llm/tests/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/air_llm/tests/test_automodel.py b/air_llm/tests/test_automodel.py
index 8224295..1b71148 100644
--- a/air_llm/tests/test_automodel.py
+++ b/air_llm/tests/test_automodel.py
@@ -1,10 +1,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