jinns 0.8.3__py3-none-any.whl → 0.8.5__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 (7) hide show
  1. jinns/loss/_DynamicLossAbstract.py +1 -1
  2. jinns/solver/_solve.py +3 -1
  3. {jinns-0.8.3.dist-info → jinns-0.8.5.dist-info}/METADATA +1 -1
  4. {jinns-0.8.3.dist-info → jinns-0.8.5.dist-info}/RECORD +7 -7
  5. {jinns-0.8.3.dist-info → jinns-0.8.5.dist-info}/LICENSE +0 -0
  6. {jinns-0.8.3.dist-info → jinns-0.8.5.dist-info}/WHEEL +0 -0
  7. {jinns-0.8.3.dist-info → jinns-0.8.5.dist-info}/top_level.txt +0 -0
jinns/loss/_DynamicLossAbstract.py CHANGED
@@ -154,7 +154,7 @@ class PDEStatio(DynamicLoss):
154
154
  _params = {
155
155
  "nn_params": params["nn_params"],
156
156
  "eq_params": self.eval_heterogeneous_parameters(
157
- t, u, params, self.eq_params_heterogeneity
157
+ x, u, params, self.eq_params_heterogeneity
158
158
  ),
159
159
  }
160
160
  new_args = args[:-1] + (_params,)
jinns/solver/_solve.py CHANGED
@@ -218,6 +218,7 @@ def solve(
218
218
  batch, data, param_data, obs_data = get_batch(data, param_data, obs_data)
219
219
 
220
220
  (
221
+ loss,
221
222
  loss_val,
222
223
  loss_terms,
223
224
  params,
@@ -311,7 +312,7 @@ def solve(
311
312
  )
312
313
 
313
314
 
314
- @partial(jit, static_argnames=["loss", "optimizer"])
315
+ @partial(jit, static_argnames=["optimizer"])
315
316
  def gradient_step(loss, optimizer, batch, params, opt_state, last_non_nan_params):
316
317
  """
317
318
  loss and optimizer cannot be jit-ted.
@@ -330,6 +331,7 @@ def gradient_step(loss, optimizer, batch, params, opt_state, last_non_nan_params
330
331
  )
331
332
 
332
333
  return (
334
+ loss,
333
335
  loss_val,
334
336
  loss_terms,
335
337
  params,
{jinns-0.8.3.dist-info → jinns-0.8.5.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: jinns
3
- Version: 0.8.3
3
+ Version: 0.8.5
4
4
  Summary: Physics Informed Neural Network with JAX
5
5
  Author-email: Hugo Gangloff <hugo.gangloff@inrae.fr>, Nicolas Jouvin <nicolas.jouvin@inrae.fr>
6
6
  Maintainer-email: Hugo Gangloff <hugo.gangloff@inrae.fr>, Nicolas Jouvin <nicolas.jouvin@inrae.fr>
{jinns-0.8.3.dist-info → jinns-0.8.5.dist-info}/RECORD RENAMED
@@ -5,7 +5,7 @@ jinns/data/_display.py,sha256=6renz4H7kHktutmLY7HM6PmxYH7cBfGHpC7GQa1Fnlk,7778
5
5
  jinns/experimental/__init__.py,sha256=3jCIy2R2i_0Erwxg-HwISdH79Nt1XCXhS9yY1F5awiY,208
6
6
  jinns/experimental/_diffrax_solver.py,sha256=sLT22byqh-6015_fhe1xtMWlFOYcCjzYKET4sLhA9R4,6818
7
7
  jinns/loss/_DynamicLoss.py,sha256=L4CVmmF0rTPbHntgqsLLHlnrlQgLHsetUocpJm7ZYag,27461
8
- jinns/loss/_DynamicLossAbstract.py,sha256=cXUJoRVphP607WPWZ5PGSHCKPCOtolB0Q6lC09PPeZY,8504
8
+ jinns/loss/_DynamicLossAbstract.py,sha256=kTQlhLx7SBuH5dIDmYaE79sVHUZt1nUFa8LxPU5IHhM,8504
9
9
  jinns/loss/_LossODE.py,sha256=sxpgiDR6mfoREuc-qe0AkirOe5K_5oblaYCnodTNxoI,21912
10
10
  jinns/loss/_LossPDE.py,sha256=_yX3R-FrAScTn9_QfVC8PfDYRE4UQ5lnzITUYgNFitA,61766
11
11
  jinns/loss/_Losses.py,sha256=XOL3MFiKEd3ndsc78Qnpi1vbgR0B2HaAWOGGW2meDM8,11190
@@ -15,7 +15,7 @@ jinns/loss/_operators.py,sha256=zDGJqYqeYH7xd-4dtGX9PS-pf0uSOpUUXGo5SVjIJ4o,1106
15
15
  jinns/solver/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
16
16
  jinns/solver/_rar.py,sha256=K-0y1-ofOAo1n_Ea3QShSGCGKVYTwiaE_Bz9-DZMJm8,14525
17
17
  jinns/solver/_seq2seq.py,sha256=FL-42hTgmVl7O3hHh1ccFVw2bT8bW82hvlDRz971Chk,5620
18
- jinns/solver/_solve.py,sha256=6kWFWpJ33uOUzZKn7gIOM7yQsVZUwSuorOWPojVeMQY,13721
18
+ jinns/solver/_solve.py,sha256=r4jn6hx7_t-Y2rBWA2npUmWWnDg4iRbgYBHZDNn9tmY,13745
19
19
  jinns/utils/__init__.py,sha256=44ms5UR6vMw3Nf6u4RCAzPFs4fom_YbBnH9mfne8m6k,313
20
20
  jinns/utils/_hyperpinn.py,sha256=Mb5d6auzFfXcA81WgjiuhDBvAypAzVOENj_gUeqz6gI,11370
21
21
  jinns/utils/_optim.py,sha256=550kxH75TL30o1iKx1swJyP0KqyUPsJ7-imL1w65Qd0,4444
@@ -24,8 +24,8 @@ jinns/utils/_save_load.py,sha256=qgZ23nUcB8-B5IZ2guuUWC4M7r5Lxd_Ms3staScdyJo,566
24
24
  jinns/utils/_spinn.py,sha256=aeIC3DBY7f_N8HABjvBNv375dMyjll3zt6KjY2bEIkM,8058
25
25
  jinns/utils/_utils.py,sha256=8dgvWXX9NT7_7-zltWp0C9tG45ZFNwXxueyxPBb4hjo,6740
26
26
  jinns/utils/_utils_uspinn.py,sha256=qcKcOw3zrwWSQyGVj6fD8c9GinHt_U6JWN_k0auTtXM,26039
27
- jinns-0.8.3.dist-info/LICENSE,sha256=BIAkGtXB59Q_BG8f6_OqtQ1BHPv60ggE9mpXJYz2dRM,11337
28
- jinns-0.8.3.dist-info/METADATA,sha256=vUEJOPFDFIZNoDAnUAzW6YbDbtKPNI5C7D5SAXCpXDg,2482
29
- jinns-0.8.3.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
30
- jinns-0.8.3.dist-info/top_level.txt,sha256=RXbkr2hzy8WBE8aiRyrJYFqn3JeMJIhMdybLjjLTB9c,6
31
- jinns-0.8.3.dist-info/RECORD,,
27
+ jinns-0.8.5.dist-info/LICENSE,sha256=BIAkGtXB59Q_BG8f6_OqtQ1BHPv60ggE9mpXJYz2dRM,11337
28
+ jinns-0.8.5.dist-info/METADATA,sha256=FrCf4ivCoMU3olFyCedvnVxCaBCiJbNfziUn1mVtyKo,2482
29
+ jinns-0.8.5.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
30
+ jinns-0.8.5.dist-info/top_level.txt,sha256=RXbkr2hzy8WBE8aiRyrJYFqn3JeMJIhMdybLjjLTB9c,6
31
+ jinns-0.8.5.dist-info/RECORD,,
{jinns-0.8.3.dist-info → jinns-0.8.5.dist-info}/LICENSE RENAMED
File without changes
{jinns-0.8.3.dist-info → jinns-0.8.5.dist-info}/WHEEL RENAMED
File without changes