diff --git a/demos/getting_started_1.py b/demos/getting_started_1.py index 142284c82909c566e9855c3215be15538dfe7b0a..8cc166660a465dcf2ea9f5d50efe81757ec86cb6 100644 --- a/demos/getting_started_1.py +++ b/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 diff --git a/demos/getting_started_2.py b/demos/getting_started_2.py index 7599c261c2a30e36b2e67d57ecc9505908070977..f695137be2645e6d155195cce32a5298848563a3 100644 --- a/demos/getting_started_2.py +++ b/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 diff --git a/demos/getting_started_3.py b/demos/getting_started_3.py index 250cc6de035e967fb2159e685e09a101f809cf64..e5d36348e4e434b63e1b2196b8886eb5b062f80b 100644 --- a/demos/getting_started_3.py +++ b/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