From 717b6ec9bb4d5f1b7f2f424abe3a0a6c324b68d8 Mon Sep 17 00:00:00 2001
From: "Alec Thomson (S&A, Kensington WA)" <alec.thomson@csiro.au>
Date: Fri, 16 Feb 2024 11:50:47 +0900
Subject: [PATCH] Finish early

---
 fixms/fix_ms_corrs.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/fixms/fix_ms_corrs.py b/fixms/fix_ms_corrs.py
index 0958c4e..a4ad072 100644
--- a/fixms/fix_ms_corrs.py
+++ b/fixms/fix_ms_corrs.py
@@ -430,6 +430,15 @@ def fix_ms_corrs(
     pol_axis = get_pol_axis(ms, feed_idx=feed_idx)
     logger.info(f"Polarization axis is {pol_axis}", ms=ms, app_params=_function_args)
 
+    # Check for pol_axis = 0deg - no correction needed
+    if pol_axis == 0 * u.deg and not fix_stokes_factor:
+        logger.critical(
+            f"Pol axis is 0deg. No correction needed. Exiting...",
+            ms=ms,
+            app_params=_function_args,
+        )
+        return
+
     # Get the data chunk by chunk and convert the correlations
     # then write them back to the MS in the 'data_column' column
     data_chunks = get_data_chunk(ms, chunksize, data_column=data_column)