remove type ignore since it is not a mock context

sjyangkevin · sjyangkevin · commit 4542f8b3b6aa · 2025-06-29T00:33:49.000-04:00
diff --git a/providers/cohere/tests/unit/cohere/operators/test_embedding.py b/providers/cohere/tests/unit/cohere/operators/test_embedding.py
@@ -53,6 +53,6 @@ def test_cohere_embedding_operator(cohere_client, get_connection):
         request_options=request_options,
     )
 
-    val = op.execute(context={})  # type: ignore[arg-type]
+    val = op.execute(context={})
     cohere_client.assert_called_once_with(api_key=api_key, base_url=base_url, timeout=timeout)
     assert val == embedded_obj

Original file line number	Diff line number	Diff line change
`@@ -53,6 +53,6 @@ def test_cohere_embedding_operator(cohere_client, get_connection):`
`53`	`53`	`request_options=request_options,`
`54`	`54`	`)`
`55`	`55`
`56`		`- val = op.execute(context={}) # type: ignore[arg-type]`
	`56`	`+ val = op.execute(context={})`
`57`	`57`	`cohere_client.assert_called_once_with(api_key=api_key, base_url=base_url, timeout=timeout)`
`58`	`58`	`assert val == embedded_obj`