Fix error again

app.py

@@ -21,7 +21,7 @@ noise_standard_deviation = st.sidebar.slider('Standard deviation of the noise',
 cost_function = st.sidebar.radio('What cost function you want to use for the fitting?', options=('RMSE-Loss', 'Huber-Loss'))
 
 # Generate random data
-X = np.column_stack((jnp.ones(number_of_observations),
+X = jnp.column_stack((jnp.ones(number_of_observations),
                      jax.random.uniform(key, shape=(number_of_observations,), minval=0., maxval=1.)))
 w = jnp.array([3.0, -20.0, 32.0])  # coefficients                                    
 X = jnp.column_stack((X, X[:,1] ** 2))   # add x**2 column