Cosmetics

demos/getting_started_1.py

@@ -21,9 +21,10 @@
 # 1D (set mode=0), 2D (mode=1), or on the sphere (mode=2)
 ###############################################################################
 
-import nifty5 as ift
 import numpy as np
 
+import nifty5 as ift
+
 
 def make_checkerboard_mask(position_space):
     # Checkerboard mask for 2D mode

demos/getting_started_2.py

@@ -21,9 +21,10 @@
 # 1D (set mode=0), 2D (mode=1), or on the sphere (mode=2)
 ###############################################################################
 
-import nifty5 as ift
 import numpy as np
 
+import nifty5 as ift
+
 
 def exposure_2d():
     # Structured exposure for 2D mode

demos/getting_started_3.py

@@ -116,10 +116,9 @@ if __name__ == '__main__':
 
     # Draw new samples to approximate the KL five times
     for i in range(5):
+        # Draw new samples and minimize KL
         KL = ift.KL_Energy(position, H, N_samples)
-        # Minimize KL
         KL, convergence = minimizer(KL)
-        # Update position
         position = KL.position
 
         # Plot current reconstruction