Skip to content

Commit 93ce6a7

Browse files
committed
fix: apply correct column name when burst dataframe is empty
1 parent 79e2d3b commit 93ce6a7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dont_fret/models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -398,7 +398,7 @@ def fret_2cde(
398398
alias="fret_2cde",
399399
) -> Bursts:
400400
if self.burst_data.is_empty():
401-
burst_data = self.burst_data.with_columns(pl.lit(None).alias("fret_2cde"))
401+
burst_data = self.burst_data.with_columns(pl.lit(None).alias(alias))
402402
return Bursts(burst_data, self.photon_data, self.metadata, self.cfg)
403403

404404
assert photons.timestamps_unit
@@ -428,7 +428,7 @@ def alex_2cde(
428428
alias="alex_2cde",
429429
) -> Bursts:
430430
if self.burst_data.is_empty():
431-
burst_data = self.burst_data.with_columns(pl.lit(None).alias("alex_2cde"))
431+
burst_data = self.burst_data.with_columns(pl.lit(None).alias(alias))
432432
return Bursts(burst_data, self.photon_data, self.metadata, self.cfg)
433433

434434
dex_streams = dex_streams if dex_streams else ["DD", "DA"]

0 commit comments

Comments
 (0)