From e3f1a063fed06c31ff7fe88861ff132228a1d7eb Mon Sep 17 00:00:00 2001 From: Steve Laskaridis Date: Sat, 21 Oct 2023 19:42:23 +0100 Subject: [PATCH] Further pylint fixes --- baselines/fjord/fjord/models.py | 2 +- baselines/fjord/fjord/od/samplers/base_sampler.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/baselines/fjord/fjord/models.py b/baselines/fjord/fjord/models.py index 14129378bdcf..0f3fc276decf 100644 --- a/baselines/fjord/fjord/models.py +++ b/baselines/fjord/fjord/models.py @@ -302,7 +302,7 @@ def get_lr_scheduler( optimiser: Optimizer, total_epochs: int, method: Optional[str] = "static", -) -> torch.optim.lr_scheduler._LRScheduler: +) -> torch.optim.lr_scheduler.LRScheduler: """Get the learning rate scheduler. :param optimiser: The optimiser for which to get the scheduler. diff --git a/baselines/fjord/fjord/od/samplers/base_sampler.py b/baselines/fjord/fjord/od/samplers/base_sampler.py index 6458e5dba887..28eac929df81 100644 --- a/baselines/fjord/fjord/od/samplers/base_sampler.py +++ b/baselines/fjord/fjord/od/samplers/base_sampler.py @@ -30,7 +30,7 @@ def prepare_sampler(self) -> None: self.widths.append(m.width) self.od_layers.append(m) - def width_sampler(self) -> Generator: + def width_sampler(self) -> Generator: # pylint: disable=no-self-use """Sample width.""" while True: yield None