linarphy/backscattering_experimental_data
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix bug if injection amplitude is float

Browse source
This commit is contained in:
linarphy 2025-07-13 02:11:09 +02:00
parent d3d16afbcf
commit 2bc6989ab8
Signed by: linarphy
GPG key ID: 0CBF02E039B4FFFF
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/backscattering_experimental_data/metadata/injection.py
View file

@ -77,7 +77,16 @@ class SinInjection(Injection):
def to_hdf5(self, store: File | Group) -> Group | File: def to_hdf5(self, store: File | Group) -> Group | File:
store.attrs["type"] = "sin" store.attrs["type"] = "sin"
if type(self.amplitude) is Quantity:
store.attrs["amplitude"] = self.amplitude.to_string() store.attrs["amplitude"] = self.amplitude.to_string()
elif type(self.amplitude) is float: # allowed (dimensionless)
store.attrs["amplitude"] = self.amplitude
else:
raise ValueError(
"amplitude must be a dimensionless quantity, not {}".format(
type(self.amplitude)
)
)
store.attrs["frequency"] = self.frequency.to_string() store.attrs["frequency"] = self.frequency.to_string()
return store return store