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

Mlflowcallback fix nonetype error (#17171)

Browse Source 
* Fix edge cases TypeError: 'NoneType' object is not callable

* fix style
This commit is contained in:
Nicolas Brousse 2022-05-16 09:18:30 -07:00 committed by GitHub
parent 95b6bef624
commit 2f611f85e2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/transformers/integrations.py
View File

@ -880,7 +880,11 @@ class MLflowCallback(TrainerCallback):
def __del__(self):
# if the previous run is not terminated correctly, the fluent API will
# not let you start a new run before the previous one is killed
if self._auto_end_run and self._ml_flow and self._ml_flow.active_run() is not None:
if (
self._auto_end_run
and callable(getattr(self._ml_flow, "active_run", None))
and self._ml_flow.active_run() is not None
):
self._ml_flow.end_run()