Further pylint fixes

adap · Oct 21, 2023 · e3f1a06 · e3f1a06
1 parent 1f3ef97
commit e3f1a06
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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