pyg-nightly 2.7.0.dev20250319__py3-none-any.whl → 2.7.0.dev20250321__py3-none-any.whl

{pyg_nightly-2.7.0.dev20250319.dist-info → pyg_nightly-2.7.0.dev20250321.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyg-nightly
-Version: 2.7.0.dev20250319
+Version: 2.7.0.dev20250321
 Summary: Graph Neural Network Library for PyTorch
 Keywords: deep-learning,pytorch,geometric-deep-learning,graph-neural-networks,graph-convolutional-networks
 Author-email: Matthias Fey <matthias@pyg.org>

{pyg_nightly-2.7.0.dev20250319.dist-info → pyg_nightly-2.7.0.dev20250321.dist-info}/RECORD

@@ -1,8 +1,8 @@
-torch_geometric/__init__.py,sha256=Dm2l52w6YLTTHnnsPMi5IfFYA4dvsihKQApFRaHM6Yg,1978
+torch_geometric/__init__.py,sha256=xrJKaQvqfmrGE_hsJP033JiezMoC4brKnhuM1h84RNM,1978
 torch_geometric/_compile.py,sha256=f-WQeH4VLi5Hn9lrgztFUCSrN_FImjhQa6BxFzcYC38,1338
 torch_geometric/_onnx.py,sha256=V9ffrIKSqhDw6xUZ12lkuSfNs48cQp2EeJ6Z19GfnVw,349
 torch_geometric/backend.py,sha256=lVaf7aLoVaB3M-UcByUJ1G4T4FOK6LXAg0CF4W3E8jo,1575
-torch_geometric/config_mixin.py,sha256=DGQXStKNiPp4iBvtx7aVoofWcUSuVKEHdG5WUL2oNJs,4230
+torch_geometric/config_mixin.py,sha256=hOTJu5LLVrEAZ6Pjt4ScLDLKv9aHbfAzF_3ufwKgO4I,4301
 torch_geometric/config_store.py,sha256=zdMzlgBpUmBkPovpYQh5fMNwTZLDq2OneqX47QEx7zk,16818
 torch_geometric/debug.py,sha256=cLyH9OaL2v7POyW-80b19w-ctA7a_5EZsS4aUF1wc2U,1295
 torch_geometric/deprecation.py,sha256=dWRymDIUkUVI2MeEmBG5WF4R6jObZeseSBV9G6FNfjc,858
@@ -633,7 +633,7 @@ torch_geometric/utils/undirected.py,sha256=H_nfpI0_WluOG6VfjPyldvcjL4w5USAKWu2x5
 torch_geometric/visualization/__init__.py,sha256=PyR_4K5SafsJrBr6qWrkjKr6GBL1b7FtZybyXCDEVwY,154
 torch_geometric/visualization/graph.py,sha256=ZuLPL92yGRi7lxlqsUPwL_EVVXF7P2kMcveTtW79vpA,4784
 torch_geometric/visualization/influence.py,sha256=CWMvuNA_Nf1sfbJmQgn58yS4OFpeKXeZPe7kEuvkUBw,477
-pyg_nightly-2.7.0.dev20250319.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
-pyg_nightly-2.7.0.dev20250319.dist-info/WHEEL,sha256=_2ozNFCLWc93bK4WKHCO-eDUENDlo-dgc9cU3qokYO4,82
-pyg_nightly-2.7.0.dev20250319.dist-info/METADATA,sha256=AY-r3MJZMmwW7ujEKaJ-6pinH_Zeq-wLAL2xNsEhHfM,63021
-pyg_nightly-2.7.0.dev20250319.dist-info/RECORD,,
+pyg_nightly-2.7.0.dev20250321.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
+pyg_nightly-2.7.0.dev20250321.dist-info/WHEEL,sha256=_2ozNFCLWc93bK4WKHCO-eDUENDlo-dgc9cU3qokYO4,82
+pyg_nightly-2.7.0.dev20250321.dist-info/METADATA,sha256=ImTB977J7eEORuQAmdHyrqi7iLsnywgCLG5v1IIA16Y,63021
+pyg_nightly-2.7.0.dev20250321.dist-info/RECORD,,

torch_geometric/__init__.py

@@ -31,7 +31,7 @@ from .lazy_loader import LazyLoader
 contrib = LazyLoader('contrib', globals(), 'torch_geometric.contrib')
 graphgym = LazyLoader('graphgym', globals(), 'torch_geometric.graphgym')
 
-__version__ = '2.7.0.dev20250319'
+__version__ = '2.7.0.dev20250321'
 
 __all__ = [
     'Index',

torch_geometric/config_mixin.py

@@ -3,6 +3,8 @@ from dataclasses import fields, is_dataclass
 from importlib import import_module
 from typing import Any, Dict
 
+from torch.nn import ModuleDict, ModuleList
+
 from torch_geometric.config_store import (
     class_from_dataclass,
     dataclass_from_class,
@@ -71,9 +73,9 @@ def _recursive_config(value: Any) -> Any:
         return value.config()
     if is_torch_instance(value, ConfigMixin):
         return value.config()
-    if isinstance(value, (tuple, list)):
+    if isinstance(value, (tuple, list, ModuleList)):
         return [_recursive_config(v) for v in value]
-    if isinstance(value, dict):
+    if isinstance(value, (dict, ModuleDict)):
         return {k: _recursive_config(v) for k, v in value.items()}
     return value