⚡️ Speed up function get_dependency_manager_installation_string by 139% in PR #208 (bump-gha-uv-version)

[codeflash-ai]

⚡️ This pull request contains optimizations for PR #208

If you approve this dependent PR, these changes will be merged into the original PR branch bump-gha-uv-version.

This PR will be automatically closed if the original PR is merged.

---

📄 139% (1.39x) speedup for get_dependency_manager_installation_string in codeflash/cli_cmds/cmd_init.py

⏱️ Runtime : 2.85 milliseconds → 1.19 millisecond (best of 114 runs)

📝 Explanation and details

Here is an optimized version of your program. The main optimizations are.

• Avoid repeated formatting of python_version_string by computing it once at module load time; the Python version doesn't change during runtime.
• Move the string templates out of the function, so they are created just once.
• Remove unnecessary usage of triple-quoted strings for templated outputs since there is no variable interpolation except one case.
• Inline the conditional return for a slightly reduced call stack.
• Use identity comparison is for Enum comparison (assuming DependencyManager is an Enum), which can be marginally faster.

Optimized Code:

What changed and why:

• Pre-calculating the version string (and Python setup string) at module load time removes a significant amount of redundant per-call formatting (was hot in profiling!).
• This also means sys.version_info is only accessed once.
• Enum comparison is done with is which is the idiomatic and fastest way for Enums.
• Templates are immediately ready, so nothing is constructed inside the function anymore; this yields maximum speedup for such a hot function; now it's a simple if/return.

This completely eliminates all expensive operations from the hot path of get_dependency_manager_installation_string.

✅ Correctness verification report:

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

🌀 Generated Regression Tests Details

from __future__ import annotations

import sys
from enum import Enum

# imports
import pytest  # used for our unit tests
from codeflash.cli_cmds.cmd_init import \
    get_dependency_manager_installation_string


class DependencyManager(Enum):
    PIP = "pip"
    POETRY = "poetry"
    PDM = "pdm"
    UV = "uv"
    CUSTOM = "custom"  # for edge cases
from codeflash.cli_cmds.cmd_init import \
    get_dependency_manager_installation_string

# unit tests

# -------- BASIC TEST CASES --------

def test_pip_manager_returns_python_setup():
    # Test that PIP returns the correct string with current Python version
    expected = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{sys.version_info.major}.{sys.version_info.minor}'"""
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.PIP)

def test_poetry_manager_returns_python_setup():
    # Test that POETRY returns the correct string with current Python version
    expected = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{sys.version_info.major}.{sys.version_info.minor}'"""
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.POETRY)

def test_pdm_manager_returns_python_setup():
    # Test that PDM returns the correct string with current Python version
    expected = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{sys.version_info.major}.{sys.version_info.minor}'"""
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.PDM)

def test_uv_manager_returns_uv_setup():
    # Test that UV returns the UV-specific string
    expected = """name: 🐍 Setup UV
        uses: astral-sh/setup-uv@v6
        with:
          enable-cache: true"""
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.UV)

# -------- EDGE TEST CASES --------

def test_custom_enum_value_returns_python_setup():
    # Test that an unknown (but valid Enum) dependency manager returns the Python setup string
    expected = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{sys.version_info.major}.{sys.version_info.minor}'"""
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.CUSTOM)




def test_enum_identity_not_value():
    # Test that passing DependencyManager.UV.value does not trigger the UV branch
    # Should raise AttributeError because string is not Enum
    with pytest.raises(AttributeError):
        get_dependency_manager_installation_string(DependencyManager.UV.value)

# -------- LARGE SCALE TEST CASES --------

@pytest.mark.parametrize("dep_manager", [DependencyManager.PIP, DependencyManager.POETRY, DependencyManager.PDM, DependencyManager.CUSTOM])
def test_bulk_python_managers_return_python_setup(dep_manager):
    # Test that all non-UV managers return the correct Python setup string
    expected = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{sys.version_info.major}.{sys.version_info.minor}'"""
    codeflash_output = get_dependency_manager_installation_string(dep_manager)

def test_many_calls_performance_and_consistency():
    # Test function consistency and performance under repeated calls
    expected_python = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{sys.version_info.major}.{sys.version_info.minor}'"""
    expected_uv = """name: 🐍 Setup UV
        uses: astral-sh/setup-uv@v6
        with:
          enable-cache: true"""
    # Alternate between managers, 500 times each
    for i in range(500):
        codeflash_output = get_dependency_manager_installation_string(DependencyManager.PIP)
        codeflash_output = get_dependency_manager_installation_string(DependencyManager.POETRY)
        codeflash_output = get_dependency_manager_installation_string(DependencyManager.PDM)
        codeflash_output = get_dependency_manager_installation_string(DependencyManager.UV)

def test_large_enum_extension():
    # Test that adding many Enum members does not affect function correctness
    # Dynamically create a large Enum with 1000 members
    LargeEnum = Enum('LargeEnum', {f'VAL_{i}': i for i in range(1000)})
    # Only DependencyManager.UV triggers the UV branch, so all others should return Python setup
    # Test a random member
    random_member = LargeEnum.VAL_42
    # Should raise AttributeError because it's not a DependencyManager
    with pytest.raises(AttributeError):
        get_dependency_manager_installation_string(random_member)

def test_all_dependency_managers_covered():
    # Ensure all DependencyManager values are tested and mapped correctly
    for manager in DependencyManager:
        if manager == DependencyManager.UV:
            expected = """name: 🐍 Setup UV
        uses: astral-sh/setup-uv@v6
        with:
          enable-cache: true"""
        else:
            expected = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{sys.version_info.major}.{sys.version_info.minor}'"""
        codeflash_output = get_dependency_manager_installation_string(manager)
# codeflash_output is used to check that the output of the original code is the same as that of the optimized code.

from __future__ import annotations

import sys
from enum import Enum

# imports
import pytest  # used for our unit tests
from codeflash.cli_cmds.cmd_init import \
    get_dependency_manager_installation_string


class DependencyManager(Enum):
    PIP = "pip"
    POETRY = "poetry"
    PDM = "pdm"
    CONDA = "conda"
    UV = "uv"
from codeflash.cli_cmds.cmd_init import \
    get_dependency_manager_installation_string

# unit tests

# --------------------------
# 1. Basic Test Cases
# --------------------------

def test_pip_manager_returns_python_setup_string():
    # Test for DependencyManager.PIP
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.PIP); result = codeflash_output
    py_version = sys.version_info
    expected = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{py_version.major}.{py_version.minor}'"""

def test_poetry_manager_returns_python_setup_string():
    # Test for DependencyManager.POETRY
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.POETRY); result = codeflash_output
    py_version = sys.version_info
    expected = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{py_version.major}.{py_version.minor}'"""

def test_pdm_manager_returns_python_setup_string():
    # Test for DependencyManager.PDM
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.PDM); result = codeflash_output
    py_version = sys.version_info
    expected = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{py_version.major}.{py_version.minor}'"""

def test_conda_manager_returns_python_setup_string():
    # Test for DependencyManager.CONDA
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.CONDA); result = codeflash_output
    py_version = sys.version_info
    expected = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{py_version.major}.{py_version.minor}'"""

def test_uv_manager_returns_uv_setup_string():
    # Test for DependencyManager.UV
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.UV); result = codeflash_output
    expected = """name: 🐍 Setup UV
        uses: astral-sh/setup-uv@v6
        with:
          enable-cache: true"""

# --------------------------
# 2. Edge Test Cases
# --------------------------





def test_all_enum_members_are_covered():
    # Ensure all enum members are handled and return a string
    for member in DependencyManager:
        codeflash_output = get_dependency_manager_installation_string(member); result = codeflash_output

# --------------------------
# 3. Large Scale Test Cases
# --------------------------

def test_many_calls_are_consistent():
    # Call the function 500 times with all valid enum values, ensure all outputs are as expected
    py_version = sys.version_info
    expected_python = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{py_version.major}.{py_version.minor}'"""
    expected_uv = """name: 🐍 Setup UV
        uses: astral-sh/setup-uv@v6
        with:
          enable-cache: true"""
    enums = [DependencyManager.PIP, DependencyManager.POETRY, DependencyManager.PDM, DependencyManager.CONDA, DependencyManager.UV]
    for i in range(500):
        for enum in enums:
            codeflash_output = get_dependency_manager_installation_string(enum); result = codeflash_output
            if enum == DependencyManager.UV:
                pass
            else:
                pass



def test_python_version_string_format():
    # Ensure the version string is always of the form 'X.Y'
    for enum in DependencyManager:
        codeflash_output = get_dependency_manager_installation_string(enum); result = codeflash_output
        if enum != DependencyManager.UV:
            version_str = result.split("python-version: ")[1].split("'")[1]
            parts = version_str.split('.')

# --------------------------
# Additional Edge: UV Case Sensitivity
# --------------------------

To edit these changes git checkout codeflash/optimize-pr208-2025-05-15T23.18.48 and push.

[codeflash-ai]

This PR has been automatically closed because the original PR #208 by misrasaurabh1 was closed.