fixes int * float casting error

BradyAJohnston · Jul 15, 2023 · 1727ed9 · 1727ed9
1 parent 2c49418
commit 1727ed9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/MolecularNodes/star.py b/MolecularNodes/star.py
@@ -59,11 +59,11 @@ def load_star_file(
 
         xyz = df[['rlnCoordinateX', 'rlnCoordinateY', 'rlnCoordinateZ']].to_numpy()
         pixel_size = df['rlnImagePixelSize'].to_numpy().reshape((-1, 1))
-        xyz *= pixel_size
+        xyz = xyz * pixel_size
         shift_column_names = ['rlnOriginXAngst', 'rlnOriginYAngst', 'rlnOriginZAngst']
         if all([col in df.columns for col in shift_column_names]):
             shifts_ang = df[shift_column_names].to_numpy()
-            xyz -= shifts_ang 
+            xyz = xyz - shifts_ang 
         euler_angles = df[['rlnAngleRot', 'rlnAngleTilt', 'rlnAnglePsi']].to_numpy()
         image_id = df['rlnMicrographName'].astype('category').cat.codes.to_numpy()