From 6dcb570e1b0c034be4371ffb4680d5cfa4e08d57 Mon Sep 17 00:00:00 2001
From: Yu Li <lyo.gavin@gmail.com>
Date: Tue, 19 Dec 2023 12:01:30 -0600
Subject: [PATCH] fix import

---
 air_llm/tests/test_automodel.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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