Safe update of params in update_grid during training.

Author: Sohaib-Ahmed21

pytorch_forecasting/layers/_kan/_kan_layer.py

@@ -233,5 +233,8 @@ def get_grid(num_interval):
             return grid
 
         grid = get_grid(num_interval)
-        self.grid.data = extend_grid(grid, k_extend=self.k)
-        self.coef.data = curve2coef(x_pos, y_eval, self.grid, self.k)
+        with torch.no_grad():
+            new_grid = extend_grid(grid, k_extend=self.k)
+            self.grid.copy_(new_grid)  
+            new_coef = curve2coef(x_pos, y_eval, self.grid, self.k)
+            self.coef.copy_(new_coef)  # this is a safe inplace copy