riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

check is added for the report_to variable in TrainingArguments (#35403)

Browse Source 
check for report_to variable is added
This commit is contained in:
alpertunga-bile 2025-01-17 16:39:32 +03:00 committed by GitHub
parent 54fd7e9260
commit df6d42a914
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/transformers/integrations/integration_utils.py
View File

@ -2157,6 +2157,17 @@ INTEGRATION_TO_CALLBACK = {
def get_reporting_integration_callbacks(report_to):
if report_to is None:
return []
if isinstance(report_to, str):
if "none" == report_to:
return []
elif "all" == report_to:
report_to = get_available_reporting_integrations()
else:
report_to = [report_to]
for integration in report_to:
if integration not in INTEGRATION_TO_CALLBACK:
raise ValueError(