Fix imports (#16)

explosion · Dec 7, 2022 · 9350148 · 9350148
1 parent 074cdb4
commit 9350148
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/spacy_loggers/clearml.py b/spacy_loggers/clearml.py
@@ -2,7 +2,7 @@
 from typing import Dict, Any, Tuple, Callable, List, Optional, IO
 import sys
 
-from spacy import util
+from spacy import Language, util
 from spacy.training.loggers import console_logger
 
 
@@ -29,7 +29,7 @@ def clearml_logger_v1(
     console = console_logger(progress_bar=False)
 
     def setup_logger(
-        nlp: "Language", stdout: IO = sys.stdout, stderr: IO = sys.stderr
+        nlp: Language, stdout: IO = sys.stdout, stderr: IO = sys.stderr
     ) -> Tuple[Callable[[Dict[str, Any]], None], Callable[[], None]]:
         config = nlp.config.interpolate()
         config_dot = util.dict_to_dot(config)

diff --git a/spacy_loggers/mlflow.py b/spacy_loggers/mlflow.py
@@ -35,7 +35,7 @@ def mlflow_logger_v1(
     console = console_logger(progress_bar=False)
 
     def setup_logger(
-        nlp: "Language", stdout: IO = sys.stdout, stderr: IO = sys.stderr
+        nlp: Language, stdout: IO = sys.stdout, stderr: IO = sys.stderr
     ) -> Tuple[Callable[[Dict[str, Any]], None], Callable[[], None]]:
         config = nlp.config.interpolate()
         config_dot = util.dict_to_dot(config)