⚡️ Speed up method PipelineBuilder.pipe by 180%

[codeflash-ai]

📄 180% (1.80x) speedup for PipelineBuilder.pipe in src/spdl/pipeline/_builder.py

⏱️ Runtime : 47.7 microseconds → 17.0 microseconds (best of 44 runs)

📝 Explanation and details

The optimization replaces an expensive isinstance(op, SupportsGetItem) check with a more efficient duck-typing approach using getattr(op, "__getitem__", None).

Key Changes:

• Replaced isinstance(op, SupportsGetItem) with getitem = getattr(op, "__getitem__", None) followed by if getitem is not None and callable(getitem)
• This avoids the overhead of Python's Abstract Base Class (ABC) machinery used by typing protocols

Why It's Faster:
The isinstance check with typing protocols like SupportsGetItem involves complex metaclass machinery and registration checks that are computationally expensive. The getattr approach directly queries the object's attribute dictionary, which is a simple hash table lookup - orders of magnitude faster.

Performance Impact:
Line profiler shows the original isinstance check took 94,999ns (72.1% of total time), while the optimized getattr + callable checks took only 1,786ns combined (4.2% of total time). This represents a ~53x improvement for this specific check, leading to the overall 179% speedup.

Test Case Performance:
This optimization is particularly effective for workloads that frequently create Pipe configurations, especially when the op parameter is typically a function/callable rather than a container with __getitem__. The duck-typing approach performs well regardless of the input type while being consistently faster than protocol checking.

✅ Correctness verification report:

Test	Status
⚙️ Existing Unit Tests	✅ 86 Passed
🌀 Generated Regression Tests	🔘 None Found
⏪ Replay Tests	🔘 None Found
🔎 Concolic Coverage Tests	🔘 None Found
📊 Tests Coverage	100.0%

⚙️ Existing Unit Tests and Runtime

Test File::Test Function	Original ⏱️	Optimized ⏱️	Speedup
spdl_unittest/dataloader/pipeline_test.py::test_pipeline_stage_hook_wrong_def1	23.7μs	8.56μs	177%✅
spdl_unittest/dataloader/pipeline_test.py::test_pipeline_stage_hook_wrong_def2	24.0μs	8.46μs	183%✅

To edit these changes git checkout codeflash/optimize-PipelineBuilder.pipe-mgrkzv0e and push.