From 03d174f961c8ee7cd3f5d38d7c7fe591f3c99a83 Mon Sep 17 00:00:00 2001 From: Janaka Abeywardhana Date: Mon, 23 Oct 2023 19:28:08 +0100 Subject: [PATCH] fix: type issue --- source/docq/support/metadata_extractors.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/source/docq/support/metadata_extractors.py b/source/docq/support/metadata_extractors.py index 62a64740..87652dfd 100644 --- a/source/docq/support/metadata_extractors.py +++ b/source/docq/support/metadata_extractors.py @@ -238,7 +238,7 @@ def __init__( if device is not None: import torch tdevice = torch.device(device) - self._model = self._model.to(self, device=tdevice) + self._model = self._model.to(device=tdevice) # type: ignore self._tokenizer = tokenizer or word_tokenize