From da9af264f8900014b53a83be3f5fa0f1937ab1d6 Mon Sep 17 00:00:00 2001
From: "Paul Gauthier (aider)" <aider@paulg.org>
Date: Wed, 2 Oct 2024 11:17:53 -0700
Subject: [PATCH] fix: import sanity_check_models in test_models.py

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

diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py
index 397b31485..4ad4cafbe 100644
--- a/tests/basic/test_models.py
+++ b/tests/basic/test_models.py
@@ -1,7 +1,7 @@
 import unittest
 from unittest.mock import MagicMock, patch
 
-from aider.models import Model, get_model_info, sanity_check_model
+from aider.models import Model, get_model_info, sanity_check_model, sanity_check_models
 
 
 class TestModels(unittest.TestCase):