Update early stopping callback

b9b8c0a5 · Gustaf Ahdritz · GitHub · e49c6dc7 · b9b8c0a5
Unverified Commit b9b8c0a5 authored Aug 06, 2022 by Gustaf Ahdritz Committed by GitHub Aug 06, 2022
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

openfold/utils/callbacks.py openfold/utils/callbacks.py +2 -2

No files found.
--- a/openfold/utils/callbacks.py
+++ b/openfold/utils/callbacks.py
@@ -6,8 +6,8 @@ class EarlyStoppingVerbose(EarlyStopping):
        The default EarlyStopping callback's verbose mode is too verbose.
        This class outputs a message only when it's getting ready to stop. 
    """
-    def _evalute_stopping_criteria(self, *args):
+    def _evalute_stopping_criteria(self, *args, **kwargs):
-        should_stop, reason = super()._evalute_stopping_criteria(*args)
+        should_stop, reason = super()._evalute_stopping_criteria(*args, **kwargs)
        if(should_stop):
            rank_zero_info(f"{reason}\n")