fix: Fixed typo in 2D ctf

README.md

@@ -16,3 +16,9 @@ And for development and testing use
 ```zsh
 pip install -e ".[dev,test]"
 ```
+
+Make sure to run tests before any commits:
+```zsh
+python -m pytest
+```
+

src/torch_fourier_filter/ctf.py

@@ -125,8 +125,9 @@ def calculate_ctf_2d(
         image_shape=image_shape,
         rfft=rfft,
         fftshift=fftshift,
-        norm=True,
+        norm=False,
     )
+
     fft_freq_grid = fft_freq_grid / einops.rearrange(pixel_size, "b -> b 1 1 1")
     fftfreq_grid_squared = fft_freq_grid**2
 
@@ -148,7 +149,7 @@ def calculate_ctf_2d(
     s2 = s**2
 
     yy2, xx2 = einops.rearrange(fftfreq_grid_squared, "b h w freq -> freq b h w")
-    xy = einops.reduce(fftfreq_grid, "b h w freq -> b h w", reduction="prod")
+    xy = einops.reduce(fft_freq_grid, "b h w freq -> b h w", reduction="prod")
     n4 = (
         einops.reduce(fftfreq_grid_squared, "b h w freq -> b h w", reduction="sum") ** 2
     )

src/torch_fourier_filter/dose_weight.py

@@ -81,7 +81,6 @@ def cumulative_dose_filter_3d(
         The dose weighting filter.
     """
     end_exposure = start_exposure + num_frames * flux
-    print(f"End exposure: {end_exposure}")
     # Get the frequency grid for 1 frame
     fft_freq_px = (
         fftfreq_grid(
@@ -103,7 +102,6 @@ def cumulative_dose_filter_3d(
     # Add small epsilon to prevent division by zero
     eps = 1e-10
     Ne = Ne.clamp(min=eps)
-    print(f"Ne min: {Ne.min()}, max: {Ne.max()}")
 
     return (
         2