small bugfix

cta-observatory · Jul 3, 2023 · 35b5c01 · 35b5c01
1 parent b92a057
commit 35b5c01
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lstchain/reco/dl1_to_dl2.py b/lstchain/reco/dl1_to_dl2.py
@@ -399,9 +399,9 @@ def build_models(filegammas, fileprotons,
 
     # Dealing with `sin_az_tel` missing data because of the former version of lstchain
     if 'sin_az_tel' not in df_gamma.columns:
-        df_gamma.sin_az_tel = np.sin(df_gamma.az_tel)
+        df_gamma['sin_az_tel'] = np.sin(df_gamma.az_tel)
     if 'sin_az_tel' not in df_proton.columns:
-        df_proton.sin_az_tel = np.sin(df_proton.az_tel)
+        df_proton['sin_az_tel'] = np.sin(df_proton.az_tel)
 
     # Training MC gammas in reduced viewcone
     src_r_min = config['train_gamma_src_r_deg'][0]

diff --git a/lstchain/scripts/lstchain_dl1_to_dl2.py b/lstchain/scripts/lstchain_dl1_to_dl2.py
@@ -102,7 +102,7 @@ def apply_to_file(filename, models_dict, output_dir, config):
 
     # Dealing with `sin_az_tel` missing data because of the former version of lstchain
     if 'sin_az_tel' not in data.columns:
-        data.sin_az_tel = np.sin(data.az_tel)
+        data['sin_az_tel'] = np.sin(data.az_tel)
 
     subarray_info = SubarrayDescription.from_hdf(filename)
     tel_id = config["allowed_tels"][0] if "allowed_tels" in config else 1