Fix bug in number of data points rounding

backend/src/logic.py

@@ -42,11 +42,12 @@ def generate_dataset(
     f = lambdify(('x1', 'x2'), function, modules='numpy')
 
     if generation_options.method == 'grid':
-        x1 = np.linspace(x1_lim[0], x1_lim[1], int(np.sqrt(generation_options.num_samples)))
-        x2 = np.linspace(x2_lim[0], x2_lim[1], int(np.sqrt(generation_options.num_samples)))
+        side_length = int(np.ceil(np.sqrt(generation_options.num_samples)))
+        x1 = np.linspace(x1_lim[0], x1_lim[1], side_length)
+        x2 = np.linspace(x2_lim[0], x2_lim[1], side_length)
         X1, X2 = np.meshgrid(x1, x2)
-        X1_flat = X1.flatten()
-        X2_flat = X2.flatten()
+        X1_flat = X1.flatten()[:generation_options.num_samples]
+        X2_flat = X2.flatten()[:generation_options.num_samples]
     elif generation_options.method == 'random':
         X1_flat = np.random.uniform(x1_lim[0], x1_lim[1], generation_options.num_samples)
         X2_flat = np.random.uniform(x2_lim[0], x2_lim[1], generation_options.num_samples)