minor bug fix

cta-observatory · Sep 11, 2024 · ffb05b0 · ffb05b0
1 parent 84c1a50
commit ffb05b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/magicctapipe/scripts/lst1_magic/semi_automatic_scripts/database_production/wobble_db.py b/magicctapipe/scripts/lst1_magic/semi_automatic_scripts/database_production/wobble_db.py
@@ -110,7 +110,7 @@ def main():
                 continue
             wobble_run_info = runs[0].split("/")[-1].split(source)[1]
             if "-W" in wobble_run_info:
-                wobble_run = wobble_run_info[2:6]
+                wobble_run = (wobble_run_info.split("W")[1])[0:4]
             else:
                 print(
                     f"No string matching for wobble offset found in the name of MAGIC files for {date_magic[i]}, run {magic_runs[j]}, {source}. Check it manually!"
@@ -123,7 +123,7 @@ def main():
             print(
                 f"More than one wobble offset value for LST run {lst_run}: check data!"
             )
-        wobble_str = "[" + "".join(str(x) for x in wobble) + "]"
+        wobble_str = "[" + ", ".join(str(x) for x in wobble) + "]"
         print(f"Wobble offset for LST run {lst_run}:", wobble_str)
         df_LST["wobble_offset"] = np.where(
             df_LST["LST1_run"] == lst_run, wobble_str, df_LST["wobble_offset"]