jinns 1.0.0__py3-none-any.whl → 1.2.0__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 (32) hide show
  1. jinns/data/_Batchs.py +4 -8
  2. jinns/data/_DataGenerators.py +532 -341
  3. jinns/loss/_DynamicLoss.py +150 -173
  4. jinns/loss/_DynamicLossAbstract.py +27 -73
  5. jinns/loss/_LossODE.py +45 -26
  6. jinns/loss/_LossPDE.py +85 -84
  7. jinns/loss/__init__.py +7 -6
  8. jinns/loss/_boundary_conditions.py +148 -279
  9. jinns/loss/_loss_utils.py +85 -58
  10. jinns/loss/_operators.py +441 -184
  11. jinns/parameters/_derivative_keys.py +487 -60
  12. jinns/plot/_plot.py +111 -98
  13. jinns/solver/_rar.py +102 -407
  14. jinns/solver/_solve.py +73 -38
  15. jinns/solver/_utils.py +122 -0
  16. jinns/utils/__init__.py +2 -0
  17. jinns/utils/_containers.py +3 -1
  18. jinns/utils/_hyperpinn.py +17 -7
  19. jinns/utils/_pinn.py +17 -27
  20. jinns/utils/_ppinn.py +227 -0
  21. jinns/utils/_save_load.py +13 -13
  22. jinns/utils/_spinn.py +24 -43
  23. jinns/utils/_types.py +1 -0
  24. jinns/utils/_utils.py +40 -12
  25. jinns-1.2.0.dist-info/AUTHORS +2 -0
  26. jinns-1.2.0.dist-info/METADATA +127 -0
  27. jinns-1.2.0.dist-info/RECORD +41 -0
  28. {jinns-1.0.0.dist-info → jinns-1.2.0.dist-info}/WHEEL +1 -1
  29. jinns-1.0.0.dist-info/METADATA +0 -84
  30. jinns-1.0.0.dist-info/RECORD +0 -38
  31. {jinns-1.0.0.dist-info → jinns-1.2.0.dist-info}/LICENSE +0 -0
  32. {jinns-1.0.0.dist-info → jinns-1.2.0.dist-info}/top_level.txt +0 -0
jinns/data/_Batchs.py CHANGED
@@ -9,19 +9,15 @@ class ODEBatch(eqx.Module):
9
9
 
10
10
 
11
11
  class PDENonStatioBatch(eqx.Module):
12
- times_x_inside_batch: (
13
- Float[Array, "batch_size dimension"] | Float[Array, "(batch_size**2) dimension"]
14
- )
15
- times_x_border_batch: (
16
- Float[Array, "border_batch_size dimension n_facets"]
17
- | Float[Array, "(border_batch_size**2) dimension n_facets"]
18
- )
12
+ domain_batch: Float[Array, "batch_size 1+dimension"]
13
+ border_batch: Float[Array, "batch_size dimension n_facets"]
14
+ initial_batch: Float[Array, "batch_size dimension"]
19
15
  param_batch_dict: dict = eqx.field(default=None)
20
16
  obs_batch_dict: dict = eqx.field(default=None)
21
17
 
22
18
 
23
19
  class PDEStatioBatch(eqx.Module):
24
- inside_batch: Float[Array, "batch_size dimension"]
20
+ domain_batch: Float[Array, "batch_size dimension"]
25
21
  border_batch: Float[Array, "batch_size dimension n_facets"]
26
22
  param_batch_dict: dict = eqx.field(default=None)
27
23
  obs_batch_dict: dict = eqx.field(default=None)