bug: move _CustomSource to private file

Author: mrava87

pylops/waveeqprocessing/_twoway.py

@@ -0,0 +1,43 @@
+from examples.seismic.utils import PointSource
+
+
+class _CustomSource(PointSource):
+    """Custom source
+
+    This class creates a Devito symbolic object that encapsulates a set of
+    sources with a user defined source signal wavelet ``wav``
+
+    Parameters
+    ----------
+    name : :obj:`str`
+        Name for the resulting symbol.
+    grid : :obj:`devito.types.grid.Grid`
+        The computational domain.
+    time_range : :obj:`examples.seismic.source.TimeAxis`
+        TimeAxis(start, step, num) object.
+    wav : :obj:`numpy.ndarray`
+        Wavelet of size
+
+    """
+
+    __rkwargs__ = PointSource.__rkwargs__ + ["wav"]
+
+    @classmethod
+    def __args_setup__(cls, *args, **kwargs):
+        kwargs.setdefault("npoint", 1)
+
+        return super().__args_setup__(*args, **kwargs)
+
+    def __init_finalize__(self, *args, **kwargs):
+        super().__init_finalize__(*args, **kwargs)
+
+        self.wav = kwargs.get("wav")
+
+        if not self.alias:
+            for p in range(kwargs["npoint"]):
+                self.data[:, p] = self.wavelet
+
+    @property
+    def wavelet(self):
+        """Return user-provided wavelet"""
+        return self.wav

pylops/waveeqprocessing/twoway.py

@@ -14,49 +14,7 @@
 if devito_message is None:
     from examples.seismic import AcquisitionGeometry, Model
     from examples.seismic.acoustic import AcousticWaveSolver
-    from examples.seismic.utils import PointSource
-
-
-class _CustomSource(PointSource):
-    """Custom source
-
-    This class creates a Devito symbolic object that encapsulates a set of
-    sources with a user defined source signal wavelet ``wav``
-
-    Parameters
-    ----------
-    name : :obj:`str`
-        Name for the resulting symbol.
-    grid : :obj:`devito.types.grid.Grid`
-        The computational domain.
-    time_range : :obj:`examples.seismic.source.TimeAxis`
-        TimeAxis(start, step, num) object.
-    wav : :obj:`numpy.ndarray`
-        Wavelet of size
-
-    """
-
-    __rkwargs__ = PointSource.__rkwargs__ + ["wav"]
-
-    @classmethod
-    def __args_setup__(cls, *args, **kwargs):
-        kwargs.setdefault("npoint", 1)
-
-        return super().__args_setup__(*args, **kwargs)
-
-    def __init_finalize__(self, *args, **kwargs):
-        super().__init_finalize__(*args, **kwargs)
-
-        self.wav = kwargs.get("wav")
-
-        if not self.alias:
-            for p in range(kwargs["npoint"]):
-                self.data[:, p] = self.wavelet
-
-    @property
-    def wavelet(self):
-        """Return user-provided wavelet"""
-        return self.wav
+    from examples.seismic.utils import _CustomSource
 
 
 class AcousticWave2D(LinearOperator):