Add redshift to the TNS table (#780)

* Add redshift information in the TNS table * Update the header
astrolabsoftware · Jan 15, 2024 · 4d5f922 · 4d5f922
1 parent fc3ab61
commit 4d5f922
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bin/download_tns.py b/bin/download_tns.py
@@ -1,5 +1,5 @@
 #!/usr/bin/env python
-# Copyright 2023 AstroLab Software
+# Copyright 2023-2024 AstroLab Software
 # Author: Julien Peloton
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -44,7 +44,7 @@ def format_tns_for_hbase(pdf: pd.DataFrame) -> pd.DataFrame:
     pdf_explode = pdf_val.explode('internalname')
 
     # Select columns of interest -- and create a Spark DataFrame
-    cols = ['fullname', 'ra', 'declination', 'type', 'internalname']
+    cols = ['fullname', 'ra', 'declination', 'type', 'redshift', 'internalname']
 
     return pdf_explode[cols]