ruff fix

src/refiners/foundationals/latent_diffusion/schedulers/__init__.py

@@ -1,7 +1,7 @@
 from refiners.foundationals.latent_diffusion.schedulers.ddim import DDIM
 from refiners.foundationals.latent_diffusion.schedulers.ddpm import DDPM
 from refiners.foundationals.latent_diffusion.schedulers.dpm_solver import DPMSolver
-from refiners.foundationals.latent_diffusion.schedulers.scheduler import Scheduler
 from refiners.foundationals.latent_diffusion.schedulers.euler import EulerScheduler
+from refiners.foundationals.latent_diffusion.schedulers.scheduler import Scheduler
 
 __all__ = ["Scheduler", "DPMSolver", "DDPM", "DDIM", "EulerScheduler"]

src/refiners/foundationals/latent_diffusion/schedulers/ddim.py

@@ -1,4 +1,4 @@
-from torch import Tensor, arange, device as Device, dtype as Dtype, float32, sqrt, tensor, Generator
+from torch import Generator, Tensor, arange, device as Device, dtype as Dtype, float32, sqrt, tensor
 
 from refiners.foundationals.latent_diffusion.schedulers.scheduler import NoiseSchedule, Scheduler
 

src/refiners/foundationals/latent_diffusion/schedulers/dpm_solver.py

@@ -1,8 +1,10 @@
-from refiners.foundationals.latent_diffusion.schedulers.scheduler import NoiseSchedule, Scheduler
-import numpy as np
-from torch import Tensor, device as Device, tensor, exp, float32, dtype as Dtype, Generator
 from collections import deque
 
+import numpy as np
+from torch import Generator, Tensor, device as Device, dtype as Dtype, exp, float32, tensor
+
+from refiners.foundationals.latent_diffusion.schedulers.scheduler import NoiseSchedule, Scheduler
+
 
 class DPMSolver(Scheduler):
     """Implements DPM-Solver++ from https://arxiv.org/abs/2211.01095

src/refiners/foundationals/latent_diffusion/schedulers/euler.py

@@ -1,7 +1,8 @@
-from refiners.foundationals.latent_diffusion.schedulers.scheduler import NoiseSchedule, Scheduler
-from torch import Tensor, device as Device, dtype as Dtype, float32, tensor, Generator
-import torch
 import numpy as np
+import torch
+from torch import Generator, Tensor, device as Device, dtype as Dtype, float32, tensor
+
+from refiners.foundationals.latent_diffusion.schedulers.scheduler import NoiseSchedule, Scheduler
 
 
 class EulerScheduler(Scheduler):

src/refiners/foundationals/latent_diffusion/schedulers/scheduler.py

@@ -1,8 +1,9 @@
 from abc import ABC, abstractmethod
 from enum import Enum
-from torch import Tensor, device as Device, dtype as DType, linspace, float32, sqrt, log, Generator
 from typing import TypeVar
 
+from torch import Generator, Tensor, device as Device, dtype as DType, float32, linspace, log, sqrt
+
 T = TypeVar("T", bound="Scheduler")
 
 

tests/foundationals/latent_diffusion/test_schedulers.py

@@ -2,7 +2,7 @@ from typing import cast
 from warnings import warn
 
 import pytest
-from torch import Tensor, allclose, device as Device, equal, randn, isclose
+from torch import Tensor, allclose, device as Device, equal, isclose, randn
 
 from refiners.fluxion import manual_seed
 from refiners.foundationals.latent_diffusion.schedulers import DDIM, DDPM, DPMSolver, EulerScheduler