pyg-nightly 2.7.0.dev20250804__py3-none-any.whl → 2.7.0.dev20250806__py3-none-any.whl

{pyg_nightly-2.7.0.dev20250804.dist-info → pyg_nightly-2.7.0.dev20250806.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyg-nightly
-Version: 2.7.0.dev20250804
+Version: 2.7.0.dev20250806
 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.dev20250804.dist-info → pyg_nightly-2.7.0.dev20250806.dist-info}/RECORD

@@ -1,4 +1,4 @@
-torch_geometric/__init__.py,sha256=2Kwqg0R6JVLum3ZGoay3DoRzb490O-Q5Q4uNMHoJP8I,2250
+torch_geometric/__init__.py,sha256=-SBUSK-uzEZeN8Z6cKLkB8FbFo6B3slnSZkZGbRYDaA,2250
 torch_geometric/_compile.py,sha256=9yqMTBKatZPr40WavJz9FjNi7pQj8YZAZOyZmmRGXgc,1351
 torch_geometric/_onnx.py,sha256=V9ffrIKSqhDw6xUZ12lkuSfNs48cQp2EeJ6Z19GfnVw,349
 torch_geometric/backend.py,sha256=lVaf7aLoVaB3M-UcByUJ1G4T4FOK6LXAg0CF4W3E8jo,1575
@@ -95,7 +95,7 @@ torch_geometric/datasets/gdelt_lite.py,sha256=zE1WagpgmsQARQhEgdCBtALRKyuQvIZqxT
 torch_geometric/datasets/ged_dataset.py,sha256=dtd-C6pCygNHLXgVfg3ZTWtTVHKT13Q3GlGrze1_rpo,9551
 torch_geometric/datasets/gemsec.py,sha256=oMTSryTgyed9z_4ydg3ql12KM-_35uqL1AoNls5nG8M,2820
 torch_geometric/datasets/geometry.py,sha256=-BxUMirZcUOf01c3avvF0b6wGPn-4S3Zj3Oau1RaJVk,4223
-torch_geometric/datasets/git_mol_dataset.py,sha256=l5u4U86tfjJdHtQPN7SM3Yjv25LD1Idtm7VHaqJqNik,10665
+torch_geometric/datasets/git_mol_dataset.py,sha256=dIn49U6gU2tPM7SrPyOkqoxJT1FhFGMmh9vdWnD30pY,10695
 torch_geometric/datasets/github.py,sha256=Qhqhkvi6eZ8VF_HqP1rL2iYToZavFNsQh7J1WdeM9dA,2687
 torch_geometric/datasets/gnn_benchmark_dataset.py,sha256=4P8n7czF-gf1egLYlAcSSvfB0GXIKpAbH5UjsuFld1M,6976
 torch_geometric/datasets/heterophilous_graph_dataset.py,sha256=yHHtwl4uPrid0vPOxvPV3sIS8HWdswar8FJ0h0OQ9is,4224
@@ -141,7 +141,7 @@ torch_geometric/datasets/polblogs.py,sha256=IYzsvd4R0OojmOOZUoOdCwQYfwcTfth1PNtc
 torch_geometric/datasets/ppi.py,sha256=zPtg-omC7WYvr9Tzwkb7zNjpXLODsvxKxKdGEUswp2E,5030
 torch_geometric/datasets/protein_mpnn_dataset.py,sha256=TTeTVJMo0Rlt2_h9bbZMKJe3rTJcjCgY5cXGyWteBfA,17756
 torch_geometric/datasets/qm7.py,sha256=bYyK8xlh9kTr5vqueNbLu9EAjIXkQH1KX1VWnjKfOJc,3323
-torch_geometric/datasets/qm9.py,sha256=Ub1t8KNeWFZvw50_Qk-80yNFeYFDwdAeyQtp3JHZs7o,17197
+torch_geometric/datasets/qm9.py,sha256=TUDQn8PkBxySNx7S2tMVzsIKYc77pJ-VfGcFoxQcpWQ,17227
 torch_geometric/datasets/rcdd.py,sha256=gvOoM1tw_X5QMyBB4FkMUwNErMXAvImyjz5twktBAh8,5317
 torch_geometric/datasets/reddit.py,sha256=QUgiKTaj6YTOYbgWgqV8mPYsctOui2ujaM8f8qy81v0,3131
 torch_geometric/datasets/reddit2.py,sha256=WSdrhbDPcUEG37XWNUd0uKnqgI911MOcfjXmgjbTPoQ,4291
@@ -517,7 +517,7 @@ torch_geometric/profile/benchmark.py,sha256=EuD12qJiiPCSwkg5w8arELXiRT_QY_3Wz_rq
 torch_geometric/profile/nvtx.py,sha256=AKBr-rqlHDnls_UM02Dfq5BZmyFTHS5Li5gaeKmsAJI,2032
 torch_geometric/profile/profile.py,sha256=cHCY4U0XtyqyKC5u380q6TspsOZ5tGHNXaZsKuzYi1A,11793
 torch_geometric/profile/profiler.py,sha256=rfNciRzWDka_BgO6aPFi3cy8mcT4lSgFWy-WfPgI2SI,16891
-torch_geometric/profile/utils.py,sha256=7h6vzTzW8vv-ZqMOz2DV8HHNgC9ViOrN7IR9d3BPDZ8,5497
+torch_geometric/profile/utils.py,sha256=ynlUVemNJZ6XjJKIkPJNwFPoUyBgVAnchfHBpKOp_HE,5903
 torch_geometric/sampler/__init__.py,sha256=0h_xJ7CQnlTxF5hUpc81WPQ0QaBtouG8eKK1RzPGA-s,512
 torch_geometric/sampler/base.py,sha256=T7RMx14RSlEKlQUkMvR1EzREaXi14VgR5GIwLfvbXzQ,27055
 torch_geometric/sampler/hgt_sampler.py,sha256=jizRJyEoz4WBOEELuqdytG2hB3UpVQX7yVPM83kvpfE,2991
@@ -639,13 +639,13 @@ torch_geometric/utils/num_nodes.py,sha256=F15ciTFOe8AxjkUh1wKH7RLmJvQYYpz-l3pPPv
 torch_geometric/utils/ppr.py,sha256=ebiHbQqRJsQbGUI5xu-IkzQSQsgIaC71vgO0KcXIKAk,4055
 torch_geometric/utils/random.py,sha256=Rv5HlhG5310rytbT9EZ7xWLGKQfozfz1azvYi5nx2-U,5148
 torch_geometric/utils/repeat.py,sha256=RxCoRoEisaP6NouXPPW5tY1Rn-tIfrmpJPm0qGP6W8M,815
-torch_geometric/utils/smiles.py,sha256=CFqeNtSBXQtY9Ex2gQzI0La490IpVVrm01QdRYEpV7w,7114
+torch_geometric/utils/smiles.py,sha256=3LBvoy9N5ZA-tLjXDUDasNPkyEvL_VyIjRvS_g4MRt0,7144
 torch_geometric/utils/sparse.py,sha256=1DbaEwdyvnzvg5qVjPlnWcEVDMkxrQLX1jJ0dr6P4js,25135
 torch_geometric/utils/undirected.py,sha256=H_nfpI0_WluOG6VfjPyldvcjL4w5USAKWu2x5nUAUjw,6222
 torch_geometric/visualization/__init__.py,sha256=b-HnVesXjyJ_L1N-DnjiRiRVf7lhwKaBQF_2i5YMVSU,208
 torch_geometric/visualization/graph.py,sha256=mfZHXYfiU-CWMtfawYc80IxVwVmtK9hbIkSKhM_j7oI,14311
 torch_geometric/visualization/influence.py,sha256=CWMvuNA_Nf1sfbJmQgn58yS4OFpeKXeZPe7kEuvkUBw,477
-pyg_nightly-2.7.0.dev20250804.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
-pyg_nightly-2.7.0.dev20250804.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-pyg_nightly-2.7.0.dev20250804.dist-info/METADATA,sha256=tiDz9kI7m8FWQ0EBKzSUvDXfi0LApo_zENmf8VCjWaI,63484
-pyg_nightly-2.7.0.dev20250804.dist-info/RECORD,,
+pyg_nightly-2.7.0.dev20250806.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
+pyg_nightly-2.7.0.dev20250806.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+pyg_nightly-2.7.0.dev20250806.dist-info/METADATA,sha256=IY8ra_cU1lz1W4NbKFu4RGfIO_uMxvpzrrAkn4CQr28,63484
+pyg_nightly-2.7.0.dev20250806.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.dev20250804'
+__version__ = '2.7.0.dev20250806'
 
 __all__ = [
     'Index',

torch_geometric/datasets/git_mol_dataset.py

@@ -102,7 +102,7 @@ class GitMolDataset(InMemoryDataset):
 
         try:
             from rdkit import Chem, RDLogger
-            RDLogger.DisableLog('rdApp.*')
+            RDLogger.DisableLog('rdApp.*')  # type: ignore[attr-defined]
             WITH_RDKIT = True
 
         except ImportError:

torch_geometric/datasets/qm9.py

@@ -202,7 +202,7 @@ class QM9(InMemoryDataset):
             from rdkit import Chem, RDLogger
             from rdkit.Chem.rdchem import BondType as BT
             from rdkit.Chem.rdchem import HybridizationType
-            RDLogger.DisableLog('rdApp.*')
+            RDLogger.DisableLog('rdApp.*')  # type: ignore[attr-defined]
             WITH_RDKIT = True
 
         except ImportError:

torch_geometric/profile/utils.py

@@ -119,19 +119,35 @@ def get_gpu_memory_from_nvidia_smi(  # pragma: no cover
         digits (int): The number of decimals to use for megabytes.
             (default: :obj:`2`)
     """
+    def parse_memory(output: str) -> list:
+        lines = output.decode('utf-8').split('\n')[1:-1]
+        mem_list = []
+        for line in lines:
+            val = line.split()[0]
+            if val != '[N/A]':
+                mem_list.append(int(val))
+            else:
+                mem_list.append(None)
+        return mem_list
+
+    def get_gpu_memory(out_device, digits):
+        if out_device is None:
+            return 0
+
+        return medibyte_to_megabyte(out_device, digits)
+
     CMD = 'nvidia-smi --query-gpu=memory.free --format=csv'
-    free_out = sp.check_output(CMD.split()).decode('utf-8').split('\n')[1:-1]
+    free_out = parse_memory(sp.check_output(CMD.split()))
 
     CMD = 'nvidia-smi --query-gpu=memory.used --format=csv'
-    used_out = sp.check_output(CMD.split()).decode('utf-8').split('\n')[1:-1]
+    used_out = parse_memory(sp.check_output(CMD.split()))
 
     if device < 0 or device >= len(free_out):
         raise AttributeError(
             f'GPU {device} not available (found {len(free_out)} GPUs)')
 
-    free_mem = medibyte_to_megabyte(int(free_out[device].split()[0]), digits)
-    used_mem = medibyte_to_megabyte(int(used_out[device].split()[0]), digits)
-
+    free_mem = get_gpu_memory(free_out[device], digits)
+    used_mem = get_gpu_memory(used_out[device], digits)
     return free_mem, used_mem
 
 

torch_geometric/utils/smiles.py

@@ -148,7 +148,7 @@ def from_smiles(
     """
     from rdkit import Chem, RDLogger
 
-    RDLogger.DisableLog('rdApp.*')
+    RDLogger.DisableLog('rdApp.*')  # type: ignore[attr-defined]
 
     mol = Chem.MolFromSmiles(smiles)