ml4gw 0.7.0__tar.gz → 0.7.1__tar.gz

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 (55) hide show
  1. {ml4gw-0.7.0 → ml4gw-0.7.1}/PKG-INFO +1 -1
  2. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/waveforms/generator.py +2 -2
  3. {ml4gw-0.7.0 → ml4gw-0.7.1}/pyproject.toml +1 -1
  4. {ml4gw-0.7.0 → ml4gw-0.7.1}/LICENSE +0 -0
  5. {ml4gw-0.7.0 → ml4gw-0.7.1}/README.md +0 -0
  6. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/__init__.py +0 -0
  7. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/augmentations.py +0 -0
  8. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/constants.py +0 -0
  9. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/dataloading/__init__.py +0 -0
  10. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/dataloading/chunked_dataset.py +0 -0
  11. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/dataloading/hdf5_dataset.py +0 -0
  12. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/dataloading/in_memory_dataset.py +0 -0
  13. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/distributions.py +0 -0
  14. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/gw.py +0 -0
  15. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/__init__.py +0 -0
  16. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/autoencoder/__init__.py +0 -0
  17. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/autoencoder/base.py +0 -0
  18. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/autoencoder/convolutional.py +0 -0
  19. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/autoencoder/skip_connection.py +0 -0
  20. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/autoencoder/utils.py +0 -0
  21. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/norm.py +0 -0
  22. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/resnet/__init__.py +0 -0
  23. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/resnet/resnet_1d.py +0 -0
  24. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/resnet/resnet_2d.py +0 -0
  25. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/streaming/__init__.py +0 -0
  26. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/streaming/online_average.py +0 -0
  27. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/streaming/snapshotter.py +0 -0
  28. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/spectral.py +0 -0
  29. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/transforms/__init__.py +0 -0
  30. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/transforms/iirfilter.py +0 -0
  31. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/transforms/pearson.py +0 -0
  32. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/transforms/qtransform.py +0 -0
  33. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/transforms/scaler.py +0 -0
  34. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/transforms/snr_rescaler.py +0 -0
  35. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/transforms/spectral.py +0 -0
  36. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/transforms/spectrogram.py +0 -0
  37. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/transforms/spline_interpolation.py +0 -0
  38. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/transforms/transform.py +0 -0
  39. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/transforms/waveforms.py +0 -0
  40. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/transforms/whitening.py +0 -0
  41. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/types.py +0 -0
  42. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/utils/interferometer.py +0 -0
  43. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/utils/slicing.py +0 -0
  44. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/waveforms/__init__.py +0 -0
  45. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/waveforms/adhoc/__init__.py +0 -0
  46. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/waveforms/adhoc/ringdown.py +0 -0
  47. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/waveforms/adhoc/sine_gaussian.py +0 -0
  48. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/waveforms/cbc/__init__.py +0 -0
  49. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/waveforms/cbc/coefficients.py +0 -0
  50. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/waveforms/cbc/phenom_d.py +0 -0
  51. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/waveforms/cbc/phenom_d_data.py +0 -0
  52. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/waveforms/cbc/phenom_p.py +0 -0
  53. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/waveforms/cbc/taylorf2.py +0 -0
  54. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/waveforms/cbc/utils.py +0 -0
  55. {ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/waveforms/conversion.py +0 -0
{ml4gw-0.7.0 → ml4gw-0.7.1}/PKG-INFO 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 → ml4gw-0.7.1}/ml4gw/waveforms/generator.py RENAMED
@@ -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 → ml4gw-0.7.1}/pyproject.toml RENAMED
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "ml4gw"
3
- version = "0.7.0"
3
+ version = "0.7.1"
4
4
  description = "Tools for training torch models on gravitational wave data"
5
5
  readme = "README.md"
6
6
  authors = [
{ml4gw-0.7.0 → ml4gw-0.7.1}/LICENSE RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/README.md RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/__init__.py RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/augmentations.py RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/constants.py RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/distributions.py RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/gw.py RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/__init__.py RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/norm.py RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/nn/resnet/__init__.py RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/spectral.py RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/transforms/pearson.py RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/transforms/scaler.py RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/types.py RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/utils/slicing.py RENAMED
File without changes
{ml4gw-0.7.0 → ml4gw-0.7.1}/ml4gw/waveforms/__init__.py RENAMED
File without changes