Use ALIAS when selecting casted columns

astrolabsoftware · Nov 20, 2023 · 048d87b · 048d87b
1 parent de6a9f2
commit 048d87b
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/bin/index_archival.py b/bin/index_archival.py
@@ -91,12 +91,12 @@ def main():
     # Restrict the input DataFrame to the subset of wanted columns.
     if 'upper' in args.index_table:
         df = data.select(
-            F.col('objectId').cast('string'),
-            F.col('prv_candidates.jd').cast('array<double>'),
-            F.col('prv_candidates.fid').cast('array<int>'),
-            F.col('prv_candidates.magpsf').cast('array<float>'),
-            F.col('prv_candidates.sigmapsf').cast('array<float>'),
-            F.col('prv_candidates.diffmaglim').cast('array<float>')
+            F.col('objectId').cast('string').alias('objectId'),
+            F.col('prv_candidates.jd').cast('array<double>').alias('jd'),
+            F.col('prv_candidates.fid').cast('array<int>').alias('fid'),
+            F.col('prv_candidates.magpsf').cast('array<float>').alias('magpsf'),
+            F.col('prv_candidates.sigmapsf').cast('array<float>').alias('sigmapsf'),
+            F.col('prv_candidates.diffmaglim').cast('array<float>').alias('diffmaglim')
         )
     else:
         df = df_flat