diff --git a/tests/basic/test_models.py b/tests/basic/test_models.py index c08784eb8..dbfd77503 100644 --- a/tests/basic/test_models.py +++ b/tests/basic/test_models.py @@ -101,10 +101,10 @@ class TestModels(unittest.TestCase): test_settings = [ { "name": "aider/extra", - "extra_params" : { - "extra_headers" : { "Foo" : "bar" }, - "some_param" : "some value", - } + "extra_params": { + "extra_headers": {"Foo": "bar"}, + "some_param": "some value", + }, }, ] @@ -133,6 +133,5 @@ class TestModels(unittest.TestCase): self.assertEqual(model.extra_params["some_param"], "some value") - if __name__ == "__main__": unittest.main()