style fix

Serenagu525 · Serenagu525 · commit 8df8dbd6d00a · 2025-07-25T21:23:56.000Z
diff --git a/src/maxdiffusion/schedulers/scheduling_rectified_flow.py b/src/maxdiffusion/schedulers/scheduling_rectified_flow.py
@@ -167,9 +167,9 @@ class FlaxRectifiedFlowSchedulerOutput(FlaxSchedulerOutput):
 
 
 class FlaxRectifiedFlowMultistepScheduler(FlaxSchedulerMixin, ConfigMixin):
-  '''
+  """
   Note: shifting and stochastic sampling not tested
-  '''
+  """
 
   dtype: jnp.dtype
   order = 1
