Move TokenAuth import out of global scope

ecmwf · Aug 1, 2024 · af5bf7a · af5bf7a
1 parent f167df2
commit af5bf7a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/anemoi/training/commands/mlflow.py b/src/anemoi/training/commands/mlflow.py
@@ -6,8 +6,6 @@
 # nor does it submit to any jurisdiction.
 
 
-from anemoi.training.diagnostics.mlflow.auth import TokenAuth
-
 from . import Command
 
 
@@ -40,6 +38,8 @@ def add_arguments(self, command_parser):
 
     def run(self, args):
         if args.subcommand == "login":
+            from anemoi.training.diagnostics.mlflow.auth import TokenAuth
+
             TokenAuth(url=args.url).login(force_credentials=args.force_credentials)
             return