chore: address CI feedback on git worktree, bytecode cleanup, and imports

Author: hebaalazzeh

ci/run_single_test.sh

@@ -118,20 +118,29 @@ case ${TEST_TYPE} in
             BASELINE_CSV="${PROFILER_TEMP_DIR}/baseline_${PACKAGE_NAME}.csv"
 
             if [ -n "${TARGET_BRANCH}" ]; then
-                # Suppress errors if branch isn't found
-                BASELINE_COMMIT=$(git merge-base HEAD "origin/${TARGET_BRANCH:-main}" 2>/dev/null || true)
+                # Try upstream first (for forks), then origin
+                BASELINE_COMMIT=$(git merge-base HEAD "upstream/${TARGET_BRANCH}" 2>/dev/null || \
+                                 git merge-base HEAD "origin/${TARGET_BRANCH}" 2>/dev/null || \
+                                 git merge-base HEAD "${TARGET_BRANCH}" 2>/dev/null || true)
                 if [ -n "${BASELINE_COMMIT}" ]; then
-                    echo "Checking out baseline commit ${BASELINE_COMMIT}..."
-                    git checkout ${BASELINE_COMMIT}
-                    if [ -f setup.py ] || [ -f pyproject.toml ]; then
-                        pip install -e .
-                        python ${PROFILER_SCRIPT} --package ${PACKAGE_NAME} --iterations 11 --csv "${BASELINE_CSV}"
+                    echo "Checking out baseline commit ${BASELINE_COMMIT} in a temporary worktree..."
+                    REPO_PREFIX=$(git rev-parse --show-prefix)
+                    WORKTREE_DIR=$(mktemp -d)
+                    rmdir "${WORKTREE_DIR}"
+                    if git worktree add "${WORKTREE_DIR}" "${BASELINE_COMMIT}" 2>/dev/null; then
+                        (
+                            cd "${WORKTREE_DIR}/${REPO_PREFIX}"
+                            if [ -f setup.py ] || [ -f pyproject.toml ]; then
+                                pip install -e .
+                                python "${PROFILER_SCRIPT}" --package "${PACKAGE_NAME}" --iterations 11 --csv "${BASELINE_CSV}"
+                            fi
+                        )
+                        git worktree remove -f "${WORKTREE_DIR}"
                     else
-                        echo "setup.py/pyproject.toml not found on baseline. Skipping baseline generation."
+                        echo "Failed to create git worktree for baseline. Skipping baseline generation."
                     fi
-                    git checkout -
                 else
-                    echo "Could not find baseline commit for origin/${TARGET_BRANCH:-main}. Skipping baseline generation."
+                    echo "Could not find baseline commit for ${TARGET_BRANCH:-main}. Skipping baseline generation."
                 fi
             fi
 

scripts/import_profiler/profiler.py

@@ -380,6 +380,7 @@ def validate_module_name(module_name):
     def find_module_from_package(pkg):
         import importlib.metadata
         import importlib.util
+
         # 1. Try to use importlib.metadata.files (works for standard installations from PyPI/wheels)
         try:
             files = importlib.metadata.files(pkg)