jinns 1.6.0__py3-none-any.whl → 1.6.1__py3-none-any.whl

jinns/data/_DataGeneratorObservations.py

@@ -13,7 +13,7 @@ from jaxtyping import PRNGKeyArray, Int, Array, Float
 from jinns.data._Batchs import ObsBatchDict
 from jinns.data._utils import _reset_or_increment
 from jinns.data._AbstractDataGenerator import AbstractDataGenerator
-from jinns.parameters._params import EqParams
+from jinns.utils._DictToModuleMeta import DictToModuleMeta
 
 if TYPE_CHECKING:
     # imports only used in type hints
@@ -27,6 +27,15 @@ if TYPE_CHECKING:
     # before hand: this is not practical, let us not get mad at this
 
 
+class DGObservedParams(metaclass=DictToModuleMeta):
+    """
+    However, static type checkers cannot know that DGObservedParams inherit from
+    eqx.Module and explicit casting to the latter class will be needed
+    """
+
+    pass
+
+
 class DataGeneratorObservations(AbstractDataGenerator):
     r"""
     Despite the class name, it is rather a dataloader for user-provided
@@ -126,7 +135,9 @@ class DataGeneratorObservations(AbstractDataGenerator):
                     )
             # Convert the dict of observed parameters to the internal `EqParams`
             # class used by Jinns.
-            self.observed_eq_params = EqParams(observed_eq_params, "EqParams")
+            self.observed_eq_params = DGObservedParams(
+                observed_eq_params, "DGObservedParams"
+            )
         else:
             self.observed_eq_params = observed_eq_params
 

{jinns-1.6.0.dist-info → jinns-1.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jinns
-Version: 1.6.0
+Version: 1.6.1
 Summary: Physics Informed Neural Network with JAX
 Author-email: Hugo Gangloff <hugo.gangloff@inrae.fr>, Nicolas Jouvin <nicolas.jouvin@inrae.fr>
 Maintainer-email: Hugo Gangloff <hugo.gangloff@inrae.fr>, Nicolas Jouvin <nicolas.jouvin@inrae.fr>

{jinns-1.6.0.dist-info → jinns-1.6.1.dist-info}/RECORD

@@ -4,7 +4,7 @@ jinns/data/_Batchs.py,sha256=UZfJWIKAFLQJxOtWN1ybJdoSQCO6PWk3SAgV9YmNVnI,2809
 jinns/data/_CubicMeshPDENonStatio.py,sha256=dtev6j7HuVZ_IBGnKVKOlJTyZVjnW_8G1VRGpcL40VU,22922
 jinns/data/_CubicMeshPDEStatio.py,sha256=g4H9wI0FkA9sdnlSxp00QArq9IjhPrmsGgSleLdNpI4,23314
 jinns/data/_DataGeneratorODE.py,sha256=XOGeX9m5J9CkkGx2alJ5G7FZYGn5cBUaFpO-3-MKvdc,7724
-jinns/data/_DataGeneratorObservations.py,sha256=gTiilF4nTY6PDAHQc5HogMWw7g266gwnN0pxSuF6B9A,9474
+jinns/data/_DataGeneratorObservations.py,sha256=h2bS-oOUi1rYvhDHy8xT-N9voqtb_IOelL2asnXCybw,9765
 jinns/data/_DataGeneratorParameter.py,sha256=m7WCWBtmNptF8qQSWGPTQEFVzkJb8kJpidKAFBmoBiQ,10205
 jinns/data/__init__.py,sha256=DEzEmoD5TmjHCaG6pS2jmFDzuhZn1ZDpFdVa2v0jCe0,703
 jinns/data/_utils.py,sha256=xKHf7NksJ_AmrtEcpJsh7WSEvI3Yk98_cM5kmXSfmx0,5596
@@ -49,9 +49,9 @@ jinns/utils/_types.py,sha256=mmDKnZavLBRNHrfsa4w2KgZbm-3Xrlu3Vx_mJ9sKI78,1288
 jinns/utils/_utils.py,sha256=M7NXX9ok-BkH5o_xo74PB1_Cc8XiDipSl51rq82dTH4,2821
 jinns/validation/__init__.py,sha256=FTyUO-v1b8Tv-FDSQsntrH7zl9E0ENexqKMT_dFRkYo,124
 jinns/validation/_validation.py,sha256=8p6sMKiBAvA6JNm65hjkMj0997LJ0BkyCREEh0AnPVE,4803
-jinns-1.6.0.dist-info/licenses/AUTHORS,sha256=7NwCj9nU-HNG1asvy4qhQ2w7oZHrn-Lk5_wK_Ve7a3M,80
-jinns-1.6.0.dist-info/licenses/LICENSE,sha256=BIAkGtXB59Q_BG8f6_OqtQ1BHPv60ggE9mpXJYz2dRM,11337
-jinns-1.6.0.dist-info/METADATA,sha256=aRCwx1t_urD8T8szddlOYusL0KDpwvJ9Rlbzsw4AQN0,5314
-jinns-1.6.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-jinns-1.6.0.dist-info/top_level.txt,sha256=RXbkr2hzy8WBE8aiRyrJYFqn3JeMJIhMdybLjjLTB9c,6
-jinns-1.6.0.dist-info/RECORD,,
+jinns-1.6.1.dist-info/licenses/AUTHORS,sha256=7NwCj9nU-HNG1asvy4qhQ2w7oZHrn-Lk5_wK_Ve7a3M,80
+jinns-1.6.1.dist-info/licenses/LICENSE,sha256=BIAkGtXB59Q_BG8f6_OqtQ1BHPv60ggE9mpXJYz2dRM,11337
+jinns-1.6.1.dist-info/METADATA,sha256=f3RcBZX2apKaq1DNb243IOlI2ieoUr0l1a5BDw0Wfsc,5314
+jinns-1.6.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+jinns-1.6.1.dist-info/top_level.txt,sha256=RXbkr2hzy8WBE8aiRyrJYFqn3JeMJIhMdybLjjLTB9c,6
+jinns-1.6.1.dist-info/RECORD,,