fix: Catch correct ConnectionError for additional_chat_templates (#39874)

akug · web-flow · commit 01c03bf4eea3 · 2025-08-18T17:25:47.000+01:00
* fix: Catch correct ConnectionError for additional_chat_templates

* fix: don't catch timeout

* fix: formatting
diff --git a/src/transformers/utils/hub.py b/src/transformers/utils/hub.py
@@ -176,7 +176,7 @@ def list_repo_templates(
             ]
         except (GatedRepoError, RepositoryNotFoundError, RevisionNotFoundError):
             raise  # valid errors => do not catch
-        except (ConnectionError, HTTPError):
+        except (HTTPError, requests.exceptions.ConnectionError):
             pass  # offline mode, internet down, etc. => try local files
 
     # check local files

Original file line number	Diff line number	Diff line change
`@@ -176,7 +176,7 @@ def list_repo_templates(`
`176`	`176`	`]`
`177`	`177`	`except (GatedRepoError, RepositoryNotFoundError, RevisionNotFoundError):`
`178`	`178`	`raise # valid errors => do not catch`
`179`		`- except (ConnectionError, HTTPError):`
	`179`	`+ except (HTTPError, requests.exceptions.ConnectionError):`
`180`	`180`	`pass # offline mode, internet down, etc. => try local files`
`181`	`181`
`182`	`182`	`# check local files`