Update utils.py

Correcting code
ITMO-NSS-team · Dec 17, 2024 · c5c8aeb · c5c8aeb
1 parent c771e8a
commit c5c8aeb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tedeous/utils.py b/tedeous/utils.py
@@ -327,9 +327,9 @@ def init_data(grid, datapath):
     device_origin = grid.device
     grid = grid.to('cpu').detach()
 
-    test_data = np.loadtxt(datapath, comments="%", encoding='utf-8').astype(np.float32)
-    test_data = torch.from_numpy(test_data)
-    grid_data = torch.stack([coord for coord in test_data[:, :-1]])
+    init_data = np.loadtxt(datapath, comments="%", encoding='utf-8').astype(np.float32)
+    init_data = torch.from_numpy(init_data)
+    grid_data = torch.stack([coord for coord in init_data[:, :-1]])
 
     init_value = test_data[:, -1:]