frozenleaves/kernels
1
0
Fork 0
mirror of https://github.com/huggingface/kernels.git synced 2025-10-20 12:33:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add attributes to user_agent (#170)

Browse Source
This commit is contained in:
Mohamed Mekkouri
2025-10-17 17:26:19 +02:00
committed by GitHub
parent 46a2a1fd6d
commit cc754ff93e
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/kernels/utils.py
View File

@ -528,11 +528,16 @@ def _get_user_agent(
return None
if user_agent is None:
user_agent = {
"kernels": __version__,
"torch": torch.__version__,
"build_variant": build_variant(),
"file_type": "kernel",
}
user_agent = {}
if isinstance(user_agent, dict):
user_agent.update(
{
"kernels": __version__,
"torch": torch.__version__,
"build_variant": build_variant(),
"file_type": "kernel",
}
)
elif isinstance(user_agent, str):
user_agent += f"; kernels/{__version__}; torch/{torch.__version__}; build_variant/{build_variant()}; file_type/kernel"
return user_agent