This commit is contained in:
Paul Gauthier 2023-06-01 08:48:55 -07:00
parent f4bc22e4fc
commit c97a0ce30a

View file

@ -21,8 +21,8 @@ class TestMain(TestCase):
self.tempdir = tempfile.mkdtemp() self.tempdir = tempfile.mkdtemp()
os.chdir(self.tempdir) os.chdir(self.tempdir)
with patch("aider.main.Coder.check_model_availability") as mock_check: self.mock_check = patch("aider.main.Coder.check_model_availability")
mock_check.return_value = True self.mock_check.return_value = True
def tearDown(self): def tearDown(self):
os.chdir(self.original_cwd) os.chdir(self.original_cwd)