From 2d778dabb780dc3faf8b1707a29df96751776bc3 Mon Sep 17 00:00:00 2001 From: NoamKaufmanLin <86767930+NoamKaufmanLin@users.noreply.github.com> Date: Wed, 18 Jan 2023 10:14:01 +0200 Subject: [PATCH] Update qa_model.py Due to changes described on "[1.7.0] - Deprecated" section: https://pytorch-lightning.readthedocs.io/en/stable/generated/CHANGELOG.html#id42 (PyTorch-Lightning's CHANGELOG) --- qa_models/qa_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/qa_models/qa_model.py b/qa_models/qa_model.py index ac2b861..b96ea1d 100644 --- a/qa_models/qa_model.py +++ b/qa_models/qa_model.py @@ -9,7 +9,7 @@ import logging from pytorch_lightning.utilities import rank_zero_only -from pytorch_lightning.loggers import LightningLoggerBase +from pytorch_lightning.loggers.logger import Logger from qa_models.qa_utils import find_offset_index, get_token_offsets, \ find_text_start_end_indices @@ -369,5 +369,5 @@ def log_object(self, data: Dict[str, Any], step): self.logger.info(data) @classmethod - def from_logger(cls, logger: LightningLoggerBase): + def from_logger(cls, logger: Logger): return cls(logger.log_dir)