jinns 0.9.0__py3-none-any.whl → 1.0.0__py3-none-any.whl

jinns/data/__init__.py

@@ -2,14 +2,10 @@ from ._DataGenerators import (
     DataGeneratorODE,
     CubicMeshPDEStatio,
     CubicMeshPDENonStatio,
-    DataGeneratorParameter,
     DataGeneratorObservations,
+    DataGeneratorParameter,
     DataGeneratorObservationsMultiPINNs,
-    append_param_batch,
-    append_obs_batch,
-)
-from ._display import (
-    plot2d,
-    plot1d_slice,
-    plot1d_image,
 )
+from ._Batchs import ODEBatch, PDEStatioBatch, PDENonStatioBatch
+
+from ._DataGenerators import append_obs_batch, append_param_batch

jinns/experimental/__init__.py

@@ -6,5 +6,3 @@ from ._diffrax_solver import (
     neumann_boundary_condition,
     plot_diffrax_solution,
 )
-from ._sinuspinn import create_sinusPINN
-from ._spectralpinn import create_spectralPINN

jinns/experimental/_diffrax_solver.py

@@ -100,16 +100,16 @@ def reaction_diffusion_2d_vector_field(
     r"""
     Matrix stencil implementation of the reaction-diffusion equation using
     finite differences. See e.g. Section 2.6.1 of
-    https://hplgit.github.io/fdm-book/doc/pub/book/pdf/fdm-book-4screen.pdf
+    <https://hplgit.github.io/fdm-book/doc/pub/book/pdf/fdm-book-4screen.pdf>
 
     The reaction-diffusion equation is
 
-    .. math::
+    $$
         \partial_t u = D \Delta u + r u (1-u).
-
+    $$
     In diffrax, the `vector_field` is the derivative of the function y with
-    respect to time `t` in the ODE :math:`y'(t) = F(y(t), t)`. Here it does not
-    depend on `t`.
+    respect to time `t` in the ODE $y'(t) = F(y(t), t)$. Here it does not
+    depend on $t$.
     """
     D, r = args