From 5cf58f3c4b6acf67de809c1e89681aef79e85c85 Mon Sep 17 00:00:00 2001 From: zjgemi Date: Mon, 21 Oct 2024 10:12:11 +0800 Subject: [PATCH] fix pyright Signed-off-by: zjgemi --- dpgen2/exploration/render/traj_render_lammps.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dpgen2/exploration/render/traj_render_lammps.py b/dpgen2/exploration/render/traj_render_lammps.py index 4ee8cd47..d12a1f22 100644 --- a/dpgen2/exploration/render/traj_render_lammps.py +++ b/dpgen2/exploration/render/traj_render_lammps.py @@ -64,8 +64,8 @@ def _load_one_model_devi(self, fname, model_devi): else: dd = np.loadtxt(fname) if ( - len(np.shape(dd)) == 1 - ): # In case model-devi.out is 1-dimensional # type: ignore + len(np.shape(dd)) == 1 # type: ignore + ): # In case model-devi.out is 1-dimensional dd = dd.reshape((1, len(dd))) # type: ignore model_devi.add(DeviManager.MAX_DEVI_V, dd[:, 1]) # type: ignore