Update docs

docs/source/api/prox.md

@@ -0,0 +1,15 @@
+# Iterative Optimization (`deepmr.optim`)
+
+```{eval-rst}
+.. automodule:: deepmr.optim
+```
+## First-order Gradient Methods
+```{eval-rst}
+.. autosummary::
+	:toctree: generated
+	:nosignatures:
+	
+	deepmr.prox.wavelet
+```
+
+## Other Methods

docs/source/index.md

@@ -28,6 +28,7 @@ tutorials/02-dl-mri-recon.ipynb
 
 api/io.md
 api/optim.md
+api/prox.md
 
 ```
 

src/deepmr/__init__.py

@@ -4,4 +4,6 @@
 
 from . import io
 from . import optim
+from . import prox
+
 from .testdata import testdata

src/deepmr/io/image/dicom.py

@@ -13,7 +13,7 @@
 from ..utils import dicom
 from ..utils.header import Header
 
-def read_dicom(filepath: str | list | tuple):
+def read_dicom(filepath):# str | list | tuple):
     """
     Read image from dicom files.
 

src/deepmr/io/image/nifti.py

@@ -14,7 +14,7 @@
 from ..utils.header import Header
 
 
-def read_nifti(filepath: str | list | tuple):
+def read_nifti(filepath):# str | list | tuple):
     """
     Read image from nifti files.
 

src/deepmr/prox/__init__.py

@@ -5,13 +5,16 @@
 
 """
 
+from deepinv.models import BM3D, TV, TGV
+
 from . import llr as _llr
 from . import wavelet as _wavelet
 
 from .llr import * # noqa
 from .wavelet import *  # noqa
 
-__all__ = []
-__all__.extend(_admm.__all__)
-__all__.extend(_pgd.__all__)
+__all__ = ["BM3D", "TV", "TGV"]
+
+# __all__.extend(_llr.__all__)
+__all__.extend(_wavelet.__all__)