📝 Description

This PR addresses a compatibility issue between thop and recent versions of scikit-learn.
The thop library relies on older sklearn import paths that have been deprecated/removed in newer releases, causing the build/execution to fail.

🔧 Changes Made

• Patched the thop usage/imports to align with the current scikit-learn API.
• Ensured the dependency chain resolves correctly without downgrading scikit-learn explicitly (if applicable).

🎯 Context

• Issue: thop uses old sklearn methods which cause ImportError or AttributeError in modern environments.
• Fix: Updated the relevant logic to support the latest scikit-learn.

✅ Checklist

• Code runs locally.
• Synced with upstream main.