Fix pytorch lightining lint issue (mlflow#6038)

Signed-off-by: harupy <[email protected]>
postrational · Jun 8, 2022 · 13ca54e · 13ca54e
1 parent 816e035
commit 13ca54e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mlflow/pytorch/_pytorch_autolog.py b/mlflow/pytorch/_pytorch_autolog.py
@@ -370,6 +370,7 @@ def patched_fit(original, self, *args, **kwargs):
         _log_early_stop_metrics(early_stop_callback, client, run_id)
 
     if Version(pl.__version__) < Version("1.4.0"):
+        # pylint: disable-next=unexpected-keyword-arg
         summary = str(ModelSummary(self.model, mode="full"))
     else:
         summary = str(ModelSummary(self.model, max_depth=-1))