From bd8f4858dbcc608799c72193afd2203539413771 Mon Sep 17 00:00:00 2001 From: Arne Binder Date: Thu, 14 Sep 2023 17:01:57 +0200 Subject: [PATCH] adjust method name --- src/pytorch_ie/core/statistic.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/pytorch_ie/core/statistic.py b/src/pytorch_ie/core/statistic.py index 3aa02166..b1baed4a 100644 --- 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]]: