pywavelet 0.2.5__py3-none-any.whl → 0.2.7__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.
Files changed (28) hide show
  1. pywavelet/__init__.py +22 -0
  2. pywavelet/_version.py +9 -4
  3. pywavelet/backend.py +87 -17
  4. pywavelet/transforms/__init__.py +16 -4
  5. pywavelet/transforms/cupy/__init__.py +12 -0
  6. pywavelet/transforms/cupy/forward/__init__.py +3 -0
  7. pywavelet/transforms/cupy/forward/from_freq.py +92 -0
  8. pywavelet/transforms/cupy/forward/from_time.py +50 -0
  9. pywavelet/transforms/cupy/forward/main.py +106 -0
  10. pywavelet/transforms/cupy/inverse/__init__.py +3 -0
  11. pywavelet/transforms/cupy/inverse/main.py +67 -0
  12. pywavelet/transforms/cupy/inverse/to_freq.py +62 -0
  13. pywavelet/transforms/jax/__init__.py +24 -0
  14. pywavelet/transforms/jax/forward/from_freq.py +6 -0
  15. pywavelet/transforms/jax/inverse/main.py +4 -6
  16. pywavelet/transforms/jax/inverse/to_freq.py +39 -25
  17. pywavelet/transforms/numpy/forward/from_freq.py +77 -19
  18. pywavelet/transforms/numpy/forward/main.py +0 -1
  19. pywavelet/transforms/numpy/inverse/main.py +2 -4
  20. pywavelet/transforms/numpy/inverse/to_freq.py +64 -1
  21. pywavelet/transforms/phi_computer.py +68 -80
  22. pywavelet/types/plotting.py +14 -5
  23. pywavelet/types/wavelet.py +6 -6
  24. {pywavelet-0.2.5.dist-info → pywavelet-0.2.7.dist-info}/METADATA +18 -7
  25. pywavelet-0.2.7.dist-info/RECORD +43 -0
  26. {pywavelet-0.2.5.dist-info → pywavelet-0.2.7.dist-info}/WHEEL +1 -1
  27. pywavelet-0.2.5.dist-info/RECORD +0 -35
  28. {pywavelet-0.2.5.dist-info → pywavelet-0.2.7.dist-info}/top_level.txt +0 -0
{pywavelet-0.2.5.dist-info → pywavelet-0.2.7.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
- Metadata-Version: 2.1
1
+ Metadata-Version: 2.4
2
2
  Name: pywavelet
3
- Version: 0.2.5
3
+ Version: 0.2.7
4
4
  Summary: WDM wavelet transform your time/freq series!
5
5
  Author-email: Pywavelet Team <avi.vajpeyi@gmail.com>
6
6
  Project-URL: Homepage, https://pywavelet.github.io/pywavelet/
@@ -11,7 +11,7 @@ Classifier: Intended Audience :: Science/Research
11
11
  Classifier: License :: OSI Approved :: MIT License
12
12
  Classifier: Operating System :: OS Independent
13
13
  Classifier: Programming Language :: Python :: 3.8
14
- Requires-Python: >=3.8
14
+ Requires-Python: >=3.10
15
15
  Description-Content-Type: text/x-rst
16
16
  Requires-Dist: numpy
17
17
  Requires-Dist: numba
@@ -21,6 +21,12 @@ Requires-Dist: tqdm
21
21
  Requires-Dist: rich
22
22
  Requires-Dist: rocket-fft
23
23
  Requires-Dist: astropy>=5.2.1
24
+ Requires-Dist: jaxtyping
25
+ Requires-Dist: beartype
26
+ Provides-Extra: jax
27
+ Requires-Dist: jax; extra == "jax"
28
+ Provides-Extra: cupy
29
+ Requires-Dist: cupy; extra == "cupy"
24
30
  Provides-Extra: dev
25
31
  Requires-Dist: pytest>=6.0; extra == "dev"
26
32
  Requires-Dist: pytest-cov>=4.1.0; extra == "dev"
@@ -32,8 +38,7 @@ Requires-Dist: isort; extra == "dev"
32
38
  Requires-Dist: mypy; extra == "dev"
33
39
  Requires-Dist: jupyter-book; extra == "dev"
34
40
  Requires-Dist: GitPython; extra == "dev"
35
- Provides-Extra: jax
36
- Requires-Dist: jax; extra == "jax"
41
+ Requires-Dist: nbconvert; extra == "dev"
37
42
 
38
43
  pywavelet
39
44
  #########
@@ -59,17 +64,23 @@ pywavelet is available on PyPI and can be installed with `pip <https://pip.pypa.
59
64
 
60
65
  $ pip install pywavelet
61
66
 
67
+
68
+ Note: We have transforms availible in numpy, JAX and Cupy.
69
+
70
+
62
71
  For developers
63
72
  --------------
64
73
 
65
- First set up a conda environment with the latest version of python.
74
+ First set up a conda environment with python 3.10
66
75
 
67
76
  .. code-block::
68
77
 
69
- $ conda create -n pywavelet -c conda-forge python=3.12
78
+ $ mamba create -n pywavelet python=3.10
70
79
 
71
80
  .. code-block::
72
81
 
82
+ $ CONDA_OVERRIDE_CUDA=12.4 mamba install "jaxlib=*=*cuda*" jax -c conda-forge
83
+ $ CONDA_OVERRIDE_CUDA=12.4 conda install -c conda-forge cupy-core
73
84
  $ pip install -e ".[dev]"
74
85
  $ pre-commit install
75
86
 
pywavelet-0.2.7.dist-info/RECORD ADDED
@@ -0,0 +1,43 @@
1
+ pywavelet/__init__.py,sha256=K7pQ8W2w9d5qwI4KzPdTpRn5-YaUfMpjnJmg7oQnYSM,508
2
+ pywavelet/_version.py,sha256=Xk20v7uvkFqkpy9aLJzVngs1eKQn0FYUP2oyA1MEQUU,511
3
+ pywavelet/backend.py,sha256=Ixl9nVP7Lobbyff5WF-pqXo0zuM7RVbyC1j35dSO3jI,2791
4
+ pywavelet/logger.py,sha256=DyKC-pJ_N9GlVeXL00E1D8hUd8GceBg-pnn7g1YPKcM,391
5
+ pywavelet/utils.py,sha256=FqQ6V41WGHMbLC4wv_1xnwHjOPDVSWnG78sAeqbYtYU,1994
6
+ pywavelet/transforms/__init__.py,sha256=c5dnTdzm-Se3idp2LP-dHvk6fmv_ynDCixIDfLCjEPw,865
7
+ pywavelet/transforms/phi_computer.py,sha256=jVxeWtfx5P1H-_HdMsK7xHuINZAjH9bj7cA8CJ98isw,3667
8
+ pywavelet/transforms/cupy/__init__.py,sha256=8BBE6msB071WdstA860a7g64C0aHT2PZsqfEgP6nmkA,336
9
+ pywavelet/transforms/cupy/forward/__init__.py,sha256=E_A8plyfTSKDRXlAAvdiRMTe9f3Y6MbK3pXMHFg8mr0,121
10
+ pywavelet/transforms/cupy/forward/from_freq.py,sha256=GoKpg019o0Ld5QqkHOt8bpzV_w74Nii__y4tIhNcuV4,3513
11
+ pywavelet/transforms/cupy/forward/from_time.py,sha256=Og9l0KurMUdbBPyvZcP1znGimMcKBgzywF0nPLTNBes,1524
12
+ pywavelet/transforms/cupy/forward/main.py,sha256=g2Pl-j4LBg7GLlzzCSoCGuEd6NNCckJ4UvC-Pppky2w,3071
13
+ pywavelet/transforms/cupy/inverse/__init__.py,sha256=J4KIzPzbHNg_8fV_c1MpPq3slSqHQV0j3VFrjfd1Nog,121
14
+ pywavelet/transforms/cupy/inverse/main.py,sha256=5pTtGNNdwlSGDQV4sqGyzUPnmqFUgFOFUFfpqjZx07Q,1608
15
+ pywavelet/transforms/cupy/inverse/to_freq.py,sha256=gpqu5Y65ZvuET5jANp6UAuAamg2PRkpAlaAjWPh7uBk,1835
16
+ pywavelet/transforms/jax/__init__.py,sha256=_JG9EaHKD95U-Nzm5zQ1RHQ7XuBjZarMa7VpZ-y7rgY,941
17
+ pywavelet/transforms/jax/forward/__init__.py,sha256=E_A8plyfTSKDRXlAAvdiRMTe9f3Y6MbK3pXMHFg8mr0,121
18
+ pywavelet/transforms/jax/forward/from_freq.py,sha256=XYtRziPD7MCbeKf4HAucQrMzko4T0zmNV7jg5bziVwA,3910
19
+ pywavelet/transforms/jax/forward/from_time.py,sha256=4RZ8-ah0qOMP20i3-xThVWddxa1QTCvZKnGpNAJbb0g,1765
20
+ pywavelet/transforms/jax/forward/main.py,sha256=7gpHUycEclDwlb6KpLqUZoIkhJjPH0sBITBGVqepYAI,3061
21
+ pywavelet/transforms/jax/inverse/__init__.py,sha256=J4KIzPzbHNg_8fV_c1MpPq3slSqHQV0j3VFrjfd1Nog,121
22
+ pywavelet/transforms/jax/inverse/main.py,sha256=_NAoEAjjEzbIcUOEEfriycZW6_a6yhGiKdtpU52H8i8,1568
23
+ pywavelet/transforms/jax/inverse/to_freq.py,sha256=x_Apiob0ZSg7UAhDSyKPxFAZXijWABJxzla526kuD8Y,3184
24
+ pywavelet/transforms/numpy/__init__.py,sha256=1Ibsup9UwMajeZ9NCQ4BN15qZTeJ_EHkgGu8XNFdA18,255
25
+ pywavelet/transforms/numpy/forward/__init__.py,sha256=E_A8plyfTSKDRXlAAvdiRMTe9f3Y6MbK3pXMHFg8mr0,121
26
+ pywavelet/transforms/numpy/forward/from_freq.py,sha256=UNf0mLdaN9XrrHPYIiLEzdAwElReZJHzZtuykSZE9iU,4556
27
+ pywavelet/transforms/numpy/forward/from_time.py,sha256=-Y6VEKwDCYBAHAjLdO46vT-6alpM5fXTgTZ_xkYxqA8,2381
28
+ pywavelet/transforms/numpy/forward/main.py,sha256=M4ELrzDone4XVO54Yf3_khTEFjVaeykES31kwhLubyU,3949
29
+ pywavelet/transforms/numpy/inverse/__init__.py,sha256=J4KIzPzbHNg_8fV_c1MpPq3slSqHQV0j3VFrjfd1Nog,121
30
+ pywavelet/transforms/numpy/inverse/main.py,sha256=3Lxxhh9VnElToaTJHgpWE-BFzKQrGEQFupnoSgJzKfk,2972
31
+ pywavelet/transforms/numpy/inverse/to_freq.py,sha256=lgb0p0M9wuhsB1mjvzqKZXuorhMickvkqO4T0A5bE_E,4914
32
+ pywavelet/transforms/numpy/inverse/to_time.py,sha256=w5vmImdsb_4YeInZtXh0llsThLTxS0tmYDlNGJ-IUew,5080
33
+ pywavelet/types/__init__.py,sha256=5YptzQvYBnRfC8N5lpOBf9I1lzpJ0pw0QMnvIcwP3YI,122
34
+ pywavelet/types/common.py,sha256=_SMmXLrRO0Nw_A7Oa6C10kZAbj8jq9agXx7tMDjnYJg,1277
35
+ pywavelet/types/frequencyseries.py,sha256=tAbZr0vEBCe0MwH7ZjaK00UVupjRNxvjoW9LCMsiiMo,7531
36
+ pywavelet/types/plotting.py,sha256=qjv5IeuSEc9WWkfJYvz1eQRgTKTspWxj4lwB5N69SbU,11002
37
+ pywavelet/types/timeseries.py,sha256=sataMW4BPFqi23h_NBZ_U9-Svuo9pLXVRmUJI6KTXG0,9430
38
+ pywavelet/types/wavelet.py,sha256=lDhpy9bEb_I-YDQbI3elaWuU8l9E2P6wDcuAQONv8lA,13591
39
+ pywavelet/types/wavelet_bins.py,sha256=gBjhWwfjcbbSnbGZVMNUeFFVUo2DVxJS4abDUVCL7ts,1458
40
+ pywavelet-0.2.7.dist-info/METADATA,sha256=5oaGFUO6ASLeTlweptCKP5D9LUGkLQbmJNfWMVYgwxc,2571
41
+ pywavelet-0.2.7.dist-info/WHEEL,sha256=DK49LOLCYiurdXXOXwGJm6U4DkHkg4lcxjhqwRa0CP4,91
42
+ pywavelet-0.2.7.dist-info/top_level.txt,sha256=g0Ezt0Rg0X-nrd-a0pAXKVRkuWNsF2M9Ynsjb9b2UYQ,10
43
+ pywavelet-0.2.7.dist-info/RECORD,,
{pywavelet-0.2.5.dist-info → pywavelet-0.2.7.dist-info}/WHEEL RENAMED
@@ -1,5 +1,5 @@
1
1
  Wheel-Version: 1.0
2
- Generator: setuptools (75.3.0)
2
+ Generator: setuptools (78.0.2)
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any
5
5
 
pywavelet-0.2.5.dist-info/RECORD DELETED
@@ -1,35 +0,0 @@
1
- pywavelet/__init__.py,sha256=zcK3Qj4wTrGZF1rU3aT6yA9LvliAOD4DVOY7gNfHhCI,53
2
- pywavelet/_version.py,sha256=8n5F0z2KUdXF-kMA1VxiH6w0elyZUUWXosPRzNz_3I0,411
3
- pywavelet/backend.py,sha256=SmpgIBHvTO1rtIAQQN_zpVB8i6R-x23FNKJG6_JlrNs,666
4
- pywavelet/logger.py,sha256=DyKC-pJ_N9GlVeXL00E1D8hUd8GceBg-pnn7g1YPKcM,391
5
- pywavelet/utils.py,sha256=FqQ6V41WGHMbLC4wv_1xnwHjOPDVSWnG78sAeqbYtYU,1994
6
- pywavelet/transforms/__init__.py,sha256=EYX8glRWojYbrjtbgrjS4vigYTRi7FOtIV3D1UwI5fY,604
7
- pywavelet/transforms/phi_computer.py,sha256=d39RfU-7Zbdo4GnmvrV21fXnFLEtf1ccPX_mBHg96Lw,4423
8
- pywavelet/transforms/jax/__init__.py,sha256=D_f-JgFAzOIJ-EuQZhTMziD4MT6lVWS3XV9s51Cu7Kg,335
9
- pywavelet/transforms/jax/forward/__init__.py,sha256=E_A8plyfTSKDRXlAAvdiRMTe9f3Y6MbK3pXMHFg8mr0,121
10
- pywavelet/transforms/jax/forward/from_freq.py,sha256=tKEdqPyEvX8ZKVQf16wGxN3d6gkcjm_RtAHQuWHUzy4,3764
11
- pywavelet/transforms/jax/forward/from_time.py,sha256=4RZ8-ah0qOMP20i3-xThVWddxa1QTCvZKnGpNAJbb0g,1765
12
- pywavelet/transforms/jax/forward/main.py,sha256=7gpHUycEclDwlb6KpLqUZoIkhJjPH0sBITBGVqepYAI,3061
13
- pywavelet/transforms/jax/inverse/__init__.py,sha256=J4KIzPzbHNg_8fV_c1MpPq3slSqHQV0j3VFrjfd1Nog,121
14
- pywavelet/transforms/jax/inverse/main.py,sha256=-HVOOBsYo3GJvGNCsQLbNPnt9s14JvbB2bGAd9LOr3A,1647
15
- pywavelet/transforms/jax/inverse/to_freq.py,sha256=y27Lx797Hcmg3gZLS1IpoyRalmJk54my91farO6G64M,2320
16
- pywavelet/transforms/numpy/__init__.py,sha256=1Ibsup9UwMajeZ9NCQ4BN15qZTeJ_EHkgGu8XNFdA18,255
17
- pywavelet/transforms/numpy/forward/__init__.py,sha256=E_A8plyfTSKDRXlAAvdiRMTe9f3Y6MbK3pXMHFg8mr0,121
18
- pywavelet/transforms/numpy/forward/from_freq.py,sha256=JmJyjrNSb64WnpP50VZRt0BICP64iZJP5QAZTZoexkw,2675
19
- pywavelet/transforms/numpy/forward/from_time.py,sha256=-Y6VEKwDCYBAHAjLdO46vT-6alpM5fXTgTZ_xkYxqA8,2381
20
- pywavelet/transforms/numpy/forward/main.py,sha256=1YCwBuhWAPFyah-XHT5Q98rMjVtoSnaEvH3sl6NWWmA,3972
21
- pywavelet/transforms/numpy/inverse/__init__.py,sha256=J4KIzPzbHNg_8fV_c1MpPq3slSqHQV0j3VFrjfd1Nog,121
22
- pywavelet/transforms/numpy/inverse/main.py,sha256=rzPsukcvOgN4vqtTpVxNSb-60KT-L-1AYr3_OSeDulk,2988
23
- pywavelet/transforms/numpy/inverse/to_freq.py,sha256=so_TDbwdS1N8sd1QcpeAEkI10XFDtoFJGohtD4YulZM,2809
24
- pywavelet/transforms/numpy/inverse/to_time.py,sha256=w5vmImdsb_4YeInZtXh0llsThLTxS0tmYDlNGJ-IUew,5080
25
- pywavelet/types/__init__.py,sha256=5YptzQvYBnRfC8N5lpOBf9I1lzpJ0pw0QMnvIcwP3YI,122
26
- pywavelet/types/common.py,sha256=_SMmXLrRO0Nw_A7Oa6C10kZAbj8jq9agXx7tMDjnYJg,1277
27
- pywavelet/types/frequencyseries.py,sha256=tAbZr0vEBCe0MwH7ZjaK00UVupjRNxvjoW9LCMsiiMo,7531
28
- pywavelet/types/plotting.py,sha256=JNDxeP-fB8U09E90J-rVT-h5yCGA_tGRHtctbgINiRo,10625
29
- pywavelet/types/timeseries.py,sha256=sataMW4BPFqi23h_NBZ_U9-Svuo9pLXVRmUJI6KTXG0,9430
30
- pywavelet/types/wavelet.py,sha256=uHJzTS2ZXTRr7I7NHWv3qNjknSBhQUpcED3jM6ti7UM,13587
31
- pywavelet/types/wavelet_bins.py,sha256=gBjhWwfjcbbSnbGZVMNUeFFVUo2DVxJS4abDUVCL7ts,1458
32
- pywavelet-0.2.5.dist-info/METADATA,sha256=jfqO4IRwiYgyhfGvE9Wrbpw4t5P6L8kuLgAA7W82Jls,2239
33
- pywavelet-0.2.5.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
34
- pywavelet-0.2.5.dist-info/top_level.txt,sha256=g0Ezt0Rg0X-nrd-a0pAXKVRkuWNsF2M9Ynsjb9b2UYQ,10
35
- pywavelet-0.2.5.dist-info/RECORD,,