fix(lint): add E402 per-file-ignores and fix import sorting for skipped test modules

Made-with: Cursor

Author: adityamehra

instrumentation-genai/opentelemetry-instrumentation-llamaindex/tests/test_embedding_instrumentation.py

@@ -2,23 +2,25 @@
 
 import pytest
 
-pytestmark = pytest.mark.skip(
-    reason="Requires live OpenAI API key; needs VCR cassettes"
-)
-
 import os
 
 from llama_index.core import Settings
 from llama_index.core.callbacks import CallbackManager
 from llama_index.embeddings.openai import OpenAIEmbedding
+
 from opentelemetry import metrics, trace
+from opentelemetry.instrumentation.llamaindex import LlamaindexInstrumentor
 from opentelemetry.sdk.metrics import MeterProvider
 from opentelemetry.sdk.metrics.export import InMemoryMetricReader
 from opentelemetry.sdk.trace import TracerProvider
-from opentelemetry.sdk.trace.export import ConsoleSpanExporter, SimpleSpanProcessor
-
-from opentelemetry.instrumentation.llamaindex import LlamaindexInstrumentor
+from opentelemetry.sdk.trace.export import (
+    ConsoleSpanExporter,
+    SimpleSpanProcessor,
+)
 
+pytestmark = pytest.mark.skip(
+    reason="Requires live OpenAI API key; needs VCR cassettes"
+)
 
 # Global setup - shared across tests
 metric_reader = None

instrumentation-genai/opentelemetry-instrumentation-llamaindex/tests/test_llm_instrumentation.py

@@ -3,9 +3,9 @@
 import os
 
 import pytest
-
 from llama_index.core.llms import ChatMessage, MessageRole
 from llama_index.core.llms.mock import MockLLM
+
 from opentelemetry import metrics, trace
 from opentelemetry.instrumentation.llamaindex import LlamaindexInstrumentor
 from opentelemetry.sdk.metrics import MeterProvider

instrumentation-genai/opentelemetry-instrumentation-llamaindex/tests/test_rag.py

@@ -11,13 +11,10 @@
 
 import pytest
 
-pytestmark = pytest.mark.skip(
-    reason="Requires live OpenAI API key; needs VCR cassettes"
-)
-
 from llama_index.core import Document, Settings, VectorStoreIndex
 from llama_index.embeddings.openai import OpenAIEmbedding
 from llama_index.llms.openai import OpenAI
+
 from opentelemetry import trace
 from opentelemetry.instrumentation.llamaindex import LlamaindexInstrumentor
 from opentelemetry.sdk.trace import TracerProvider
@@ -27,6 +24,10 @@
     SpanExportResult,
 )
 
+pytestmark = pytest.mark.skip(
+    reason="Requires live OpenAI API key; needs VCR cassettes"
+)
+
 
 class DebugSpanExporter(SpanExporter):
     """Custom exporter that shows parent-child relationships clearly."""

instrumentation-genai/opentelemetry-instrumentation-llamaindex/tests/test_workflow_agent.py

@@ -5,24 +5,30 @@
 """
 
 import asyncio
+
 import pytest
 
-pytestmark = pytest.mark.skip(
-    reason="Event stream already consumed errors; needs rework"
-)
 from typing import List
-from llama_index.core.agent import ReActAgent
+
 from llama_index.core import Settings
-from llama_index.core.llms import MockLLM
+from llama_index.core.agent import ReActAgent
 from llama_index.core.base.llms.types import ChatMessage, MessageRole
+from llama_index.core.llms import MockLLM
 from llama_index.core.tools import FunctionTool
-from opentelemetry import trace, metrics
+
+from opentelemetry import metrics, trace
 from opentelemetry.instrumentation.llamaindex import LlamaindexInstrumentor
-from opentelemetry.sdk.trace import TracerProvider
-from opentelemetry.sdk.trace.export import SimpleSpanProcessor
-from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
 from opentelemetry.sdk.metrics import MeterProvider
 from opentelemetry.sdk.metrics.export import InMemoryMetricReader
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import SimpleSpanProcessor
+from opentelemetry.sdk.trace.export.in_memory_span_exporter import (
+    InMemorySpanExporter,
+)
+
+pytestmark = pytest.mark.skip(
+    reason="Event stream already consumed errors; needs rework"
+)
 
 
 def multiply(a: int, b: int) -> int:

instrumentation-genai/opentelemetry-instrumentation-llamaindex/tests/test_workflow_agent_variants.py

@@ -4,10 +4,11 @@
 
 import pytest
 
-pytestmark = pytest.mark.skip(
-    reason="Event stream already consumed errors; needs rework"
+from llama_index.core.agent.workflow import (
+    AgentWorkflow,
+    CodeActAgent,
+    FunctionAgent,
 )
-from llama_index.core.agent.workflow import AgentWorkflow, CodeActAgent, FunctionAgent
 from llama_index.core.base.llms.types import (
     ChatMessage,
     ChatResponse,
@@ -17,6 +18,7 @@
 from llama_index.core.llms import MockLLM
 from llama_index.core.llms.llm import ToolSelection
 from llama_index.core.workflow import StartEvent, StopEvent, Workflow, step
+
 from opentelemetry.instrumentation.llamaindex import LlamaindexInstrumentor
 from opentelemetry.sdk.metrics import MeterProvider
 from opentelemetry.sdk.trace import TracerProvider
@@ -25,6 +27,10 @@
     InMemorySpanExporter,
 )
 
+pytestmark = pytest.mark.skip(
+    reason="Event stream already consumed errors; needs rework"
+)
+
 
 class StaticChatLLM(MockLLM):
     def __init__(self, content: str):

pyproject.toml

@@ -173,6 +173,11 @@ ignore = [
 [tool.ruff.lint.per-file-ignores]
 "docs/**/*.*" = ["A001"]
 "instrumentation-genai/opentelemetry-instrumentation-langchain/tests/test_callback_handler_agent.py" = ["E402"]
+"instrumentation-genai/opentelemetry-instrumentation-llamaindex/tests/test_embedding_instrumentation.py" = ["E402"]
+"instrumentation-genai/opentelemetry-instrumentation-llamaindex/tests/test_rag.py" = ["E402"]
+"instrumentation-genai/opentelemetry-instrumentation-llamaindex/tests/test_workflow_agent.py" = ["E402"]
+"instrumentation-genai/opentelemetry-instrumentation-llamaindex/tests/test_workflow_agent_variants.py" = ["E402"]
+"instrumentation-genai/opentelemetry-instrumentation-openai-agents-v2/tests/test_tracer.py" = ["E402"]
 
 [tool.ruff.lint.isort]
 detect-same-package = false # to not consider instrumentation packages as first-party