ml4gw 0.7.0__py3-none-any.whl → 0.7.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of ml4gw might be problematic. Click here for more details.

Files changed (5) hide show
  1. ml4gw/waveforms/generator.py +2 -2
  2. {ml4gw-0.7.0.dist-info → ml4gw-0.7.1.dist-info}/METADATA +1 -1
  3. {ml4gw-0.7.0.dist-info → ml4gw-0.7.1.dist-info}/RECORD +5 -5
  4. {ml4gw-0.7.0.dist-info → ml4gw-0.7.1.dist-info}/LICENSE +0 -0
  5. {ml4gw-0.7.0.dist-info → ml4gw-0.7.1.dist-info}/WHEEL +0 -0
ml4gw/waveforms/generator.py CHANGED
@@ -224,7 +224,7 @@ class TimeDomainCBCWaveformGenerator(torch.nn.Module):
224
224
  k1s = torch.round(f_min / df)
225
225
 
226
226
  num_freqs = frequencies.size(0)
227
- frequency_indices = torch.arange(num_freqs)
227
+ frequency_indices = torch.arange(num_freqs, device=device)
228
228
  taper_mask = frequency_indices <= k1s[:, None]
229
229
  taper_mask &= frequency_indices >= k0s[:, None]
230
230
 
@@ -253,7 +253,7 @@ class TimeDomainCBCWaveformGenerator(torch.nn.Module):
253
253
  # that will translate the coalescense time such that it is `right_pad`
254
254
  # seconds from the right edge of the window
255
255
  tshift = round(self.right_pad * self.sample_rate) / self.sample_rate
256
- kvals = torch.arange(num_freqs)
256
+ kvals = torch.arange(num_freqs, device=device)
257
257
  phase_shift = torch.exp(1j * 2 * torch.pi * df * tshift * kvals)
258
258
 
259
259
  hc_spectrum *= phase_shift
{ml4gw-0.7.0.dist-info → ml4gw-0.7.1.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: ml4gw
3
- Version: 0.7.0
3
+ Version: 0.7.1
4
4
  Summary: Tools for training torch models on gravitational wave data
5
5
  Author: Alec Gunny
6
6
  Author-email: alec.gunny@ligo.org
{ml4gw-0.7.0.dist-info → ml4gw-0.7.1.dist-info}/RECORD RENAMED
@@ -48,8 +48,8 @@ ml4gw/waveforms/cbc/phenom_p.py,sha256=LBtGVUjBjROcYBPLldFnF6T1jZV6ZyuZEnkn9-oTK
48
48
  ml4gw/waveforms/cbc/taylorf2.py,sha256=2ga_lG_xkYOsF-BdxgjbU0pgLDjeAO0p5IWuCPvibvQ,10504
49
49
  ml4gw/waveforms/cbc/utils.py,sha256=CvZ79PQygb-zwulMV-wRuBcGEsHbVOtJz60UnOJFKoM,3051
50
50
  ml4gw/waveforms/conversion.py,sha256=MyADWEZVoEkRkKaHk1ZuQDsGfPYx5xUTtyApj5P3ueQ,6895
51
- ml4gw/waveforms/generator.py,sha256=yCMsL55HRYHh6AVLWKA9Pmd6AoAyO7pe1VdrxzxxOzE,11975
52
- ml4gw-0.7.0.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
53
- ml4gw-0.7.0.dist-info/METADATA,sha256=NsQW99R6S6MYDLoq_cFdgTK-lEZZBK8a-5S09RJIhZo,3904
54
- ml4gw-0.7.0.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
55
- ml4gw-0.7.0.dist-info/RECORD,,
51
+ ml4gw/waveforms/generator.py,sha256=i2lgaJzH5eA6gzc-bLQZYYEgEQ8OBLJgE9yNXU3FsKM,12005
52
+ ml4gw-0.7.1.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
53
+ ml4gw-0.7.1.dist-info/METADATA,sha256=xEVSE7PX32I8b4YIneUVVvTAHLS4WemuQ8bpCKskIXE,3904
54
+ ml4gw-0.7.1.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
55
+ ml4gw-0.7.1.dist-info/RECORD,,
{ml4gw-0.7.0.dist-info → ml4gw-0.7.1.dist-info}/LICENSE RENAMED
File without changes
{ml4gw-0.7.0.dist-info → ml4gw-0.7.1.dist-info}/WHEEL RENAMED
File without changes