adjust method name

ArneBinder · Sep 14, 2023 · bd8f485 · bd8f485
1 parent aa089f1
commit bd8f485
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/pytorch_ie/core/statistic.py b/src/pytorch_ie/core/statistic.py
@@ -5,7 +5,11 @@
 
 from pytorch_ie.core.document import Document
 from pytorch_ie.core.metric import DocumentMetric
-from pytorch_ie.utils.hydra import InstantiationException, resolve_document_type, resolve_target
+from pytorch_ie.utils.hydra import (
+    InstantiationException,
+    resolve_optional_document_type,
+    resolve_target,
+)
 
 logger = logging.getLogger(__name__)
 
@@ -162,7 +166,7 @@ def __init__(
         self.show_histogram = show_histogram
         self.show_as_markdown = show_as_markdown
         self.title = title or self.__class__.__name__
-        self._document_type = resolve_document_type(document_type)
+        self._document_type = resolve_optional_document_type(document_type)
 
     @property
     def document_type(self) -> Optional[Type[Document]]: