reformat

Signed-off-by: Can-Zhao <[email protected]>

Author: Can-Zhao

monai/networks/schedulers/rectified_flow.py

@@ -36,8 +36,8 @@
 
 from monai.utils import StrEnum
 
-from .scheduler import Scheduler
 from .ddpm import DDPMPredictionType
+from .scheduler import Scheduler
 
 
 class RFlowPredictionType(StrEnum):
@@ -46,8 +46,10 @@ class RFlowPredictionType(StrEnum):
 
     v_prediction: velocity prediction, see section 2.4 https://imagen.research.google/video/paper.pdf
     """
+
     V_PREDICTION = DDPMPredictionType.V_PREDICTION
 
+
 def timestep_transform(
     t, input_img_size_numel, base_img_size_numel=32 * 32 * 32, scale=1.0, num_train_timesteps=1000, spatial_dim=3
 ):