Apply suggestions from code review

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Jun 6, 2024 · 30b6f0f · 30b6f0f
1 parent 93d3f1d
commit 30b6f0f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/source/tests/pt/model/test_smooth_denoise.py b/source/tests/pt/model/test_smooth_denoise.py
@@ -57,7 +57,7 @@ def test(
         )
         generator = torch.Generator(device="cpu").manual_seed(GLOBAL_SEED)
         coord1 = torch.rand(
-            [natoms - coord0.shape[0], 3], dtype=dtype, genreator=generator
+            [natoms - coord0.shape[0], 3], dtype=dtype, generator=generator
         ).to(env.DEVICE)
         coord1 = torch.matmul(coord1, cell)
         coord = torch.concat([coord0, coord1], dim=0)

diff --git a/source/tests/pt/model/test_trans_denoise.py b/source/tests/pt/model/test_trans_denoise.py
@@ -37,7 +37,7 @@ def test(
         coord = torch.rand([natoms, 3], dtype=dtype).to(env.DEVICE)
         coord = torch.matmul(coord, cell)
         atype = torch.IntTensor([0, 0, 0, 1, 1]).to(env.DEVICE)
-        shift = (torch.rand([3], dtype=dtype, generaotr=generator) - 0.5).to(
+        shift = (torch.rand([3], dtype=dtype, generator=generator) - 0.5).to(
             env.DEVICE
         ) * 2.0
         coord_s = torch.matmul(