fix test import

53ec180d4079a2a84ab7c41f5568c4ac04629ccf · 2024-08-13 11:34:34 +0100 · Alex Cheema

Files touched

M exo/inference/test_inference_engine.py

Diff

commit 53ec180d4079a2a84ab7c41f5568c4ac04629ccf
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Tue Aug 13 11:34:34 2024 +0100

    fix test import
---
 exo/inference/test_inference_engine.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/exo/inference/test_inference_engine.py b/exo/inference/test_inference_engine.py
index fa228ca9..bf418e3d 100644
--- a/exo/inference/test_inference_engine.py
+++ b/exo/inference/test_inference_engine.py
@@ -3,6 +3,7 @@ from exo.download.hf.hf_shard_download import HFShardDownloader
 from exo.inference.inference_engine import InferenceEngine
 from exo.inference.shard import Shard
 from exo.helpers import DEBUG
+import os
 import asyncio
 import numpy as np

← 611085b3 trigger test · back to Exo · t a930be4f →