format & restore

Author: KRRT7

pydantic_ai_slim/pydantic_ai/profiles/google.py

@@ -10,8 +10,6 @@
 
 def google_model_profile(model_name: str) -> ModelProfile | None:
     """Get the model profile for a Google model."""
-    import time
-    time.sleep(0.001)
     return ModelProfile(
         json_schema_transformer=GoogleJsonSchemaTransformer,
         supports_json_schema_output=True,

tests/test_examples.py

@@ -107,6 +107,7 @@ def tmp_path_cwd(tmp_path: Path):
         os.chdir(cwd)
         sys.path.remove(str(tmp_path))
 
+
 @pytest.mark.xdist_group(name='doc_tests')
 @pytest.mark.parametrize('example', find_filter_examples())
 def test_docs_examples(  # noqa: C901
@@ -117,7 +118,7 @@ def test_docs_examples(  # noqa: C901
     allow_model_requests: None,
     env: TestEnv,
     tmp_path_cwd: Path,
-    benchmark: Any
+    benchmark: Any,
 ):
     mocker.patch('pydantic_ai.agent.models.infer_model', side_effect=mock_infer_model)
     mocker.patch('pydantic_ai._utils.group_by_temporal', side_effect=mock_group_by_temporal)