pyg-nightly 2.7.0.dev20251002__py3-none-any.whl → 2.7.0.dev20251004__py3-none-any.whl

{pyg_nightly-2.7.0.dev20251002.dist-info → pyg_nightly-2.7.0.dev20251004.dist-info}/METADATA

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

@@ -1,4 +1,4 @@
-torch_geometric/__init__.py,sha256=zmY8uj7LnHDgndMZ1cWvURjSKApKxsMGa7nF8KAJE0k,2292
+torch_geometric/__init__.py,sha256=yfHvrC4eQHt-2nRrRakETd_K0aEsslEH_OsrafyH7HE,2292
 torch_geometric/_compile.py,sha256=9yqMTBKatZPr40WavJz9FjNi7pQj8YZAZOyZmmRGXgc,1351
 torch_geometric/_onnx.py,sha256=ODB_8cwFUiwBUjngXn6-K5HHb7IDul7DDXuuGX7vj_0,8178
 torch_geometric/backend.py,sha256=lVaf7aLoVaB3M-UcByUJ1G4T4FOK6LXAg0CF4W3E8jo,1575
@@ -114,7 +114,7 @@ torch_geometric/datasets/linkx_dataset.py,sha256=_DsF5d2-o79-WibEKojIJKCpCF3VVxS
 torch_geometric/datasets/lrgb.py,sha256=lOlzYCn9XbwQb3HK_wdufqjqK_aZbnoUqZu0NXZ6Oyw,11657
 torch_geometric/datasets/malnet_tiny.py,sha256=E_ymC7_XS8rgZelcdevZyCDVjX5Ov21G6vwrG0JgAP0,5271
 torch_geometric/datasets/md17.py,sha256=Wv-Q75uUDrFjRur5nOvg2TSw68UxkdYDJvkf3YA-T70,16735
-torch_geometric/datasets/medshapenet.py,sha256=eCBCXKpueweCwDSf_Q4_MwVA3IbJd04FSxnknLnMHTk,5567
+torch_geometric/datasets/medshapenet.py,sha256=bd94y_bgTUkGYC0jUAg9f_NKI0x83ZNOh-SJYBEXwH0,5618
 torch_geometric/datasets/mixhop_synthetic_dataset.py,sha256=4NNvTHUvvV6pcqQCyVDS5XhppXUeF2H9GTfFoc49eyU,3951
 torch_geometric/datasets/mnist_superpixels.py,sha256=o2ArbZ0_OE0u8VCaHmWwvngESlOFr9oM9dSEP_tjAS4,3340
 torch_geometric/datasets/modelnet.py,sha256=rqR-e75lC8PS_IX7VlNbo2Az9IWfqMNvDp8rmQCp-LE,5357
@@ -271,7 +271,7 @@ torch_geometric/llm/__init__.py,sha256=qOCyBeWO8vcUTI71DLv1-lrssWoAqIyE1JTuzHM4r
 torch_geometric/llm/large_graph_indexer.py,sha256=ip5wU7Rj4S1tFc53AqAtvyvGqeVC7AJrVMEbRTHyz54,28429
 torch_geometric/llm/rag_loader.py,sha256=lsUt5kvHGbQzxOuWubBVB0RkMcTi6b4RNRuvsArw3Ys,5641
 torch_geometric/llm/models/__init__.py,sha256=Xb2GacStLOYCAu34NNzG-zog-zapXh1asl3S6_eLMsc,547
-torch_geometric/llm/models/g_retriever.py,sha256=SeW0rlrlzMhN3aVslhq_GUyUgS4sVw_nMAT5YiXzZd8,9072
+torch_geometric/llm/models/g_retriever.py,sha256=BPERbaekEyD0e7VUNy5PJlzfAqRkiYFwzWe3QEokgfs,9112
 torch_geometric/llm/models/git_mol.py,sha256=m1YJb6Xb2i6j9wEqHzqE4YBWMr9i1CJpMp6T-E24fsA,12680
 torch_geometric/llm/models/glem.py,sha256=GlL_I63g-_5eTycSGRj720YntldQ-CQ351RaDPc6XAU,16674
 torch_geometric/llm/models/llm.py,sha256=039mq9rZBZMyZW5rYj0fMP5kl9RJAI7N-oJyC4Mf2Gs,18122
@@ -526,7 +526,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=ynlUVemNJZ6XjJKIkPJNwFPoUyBgVAnchfHBpKOp_HE,5903
+torch_geometric/profile/utils.py,sha256=4Gyl3Rx5aCKJB4HkLzcL1IXAwFqHrc9Yrc7IuMWskcA,5893
 torch_geometric/sampler/__init__.py,sha256=NsAvP6S9wfR0Dsc1d5tR7fj5owRlARRUj3RuUeh8ydg,578
 torch_geometric/sampler/base.py,sha256=GifHECotqBrD82yeqFHuW67MyTB7djN3ejHEkqR3GNI,41187
 torch_geometric/sampler/hgt_sampler.py,sha256=jizRJyEoz4WBOEELuqdytG2hB3UpVQX7yVPM83kvpfE,2991
@@ -654,7 +654,7 @@ torch_geometric/utils/undirected.py,sha256=H_nfpI0_WluOG6VfjPyldvcjL4w5USAKWu2x5
 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.dev20251002.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
-pyg_nightly-2.7.0.dev20251002.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-pyg_nightly-2.7.0.dev20251002.dist-info/METADATA,sha256=A3ulW4_9Zy9-7vyoFH9h7wMI-xenp3K24p1CpTBG5p8,63680
-pyg_nightly-2.7.0.dev20251002.dist-info/RECORD,,
+pyg_nightly-2.7.0.dev20251004.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
+pyg_nightly-2.7.0.dev20251004.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+pyg_nightly-2.7.0.dev20251004.dist-info/METADATA,sha256=WZ-UTXOdyvvETpudgVsfUg3GyZsVEl1kj9XmbvEiU78,63680
+pyg_nightly-2.7.0.dev20251004.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.dev20251002'
+__version__ = '2.7.0.dev20251004'
 
 __all__ = [
     'Index',

torch_geometric/datasets/medshapenet.py

@@ -96,7 +96,8 @@ class MedShapeNet(InMemoryDataset):
 
         subset = []
         for dataset in list_of_datasets:
-            self.newpath = self.root + '/' + dataset.split("/")[1]
+            parts = dataset.split("/")
+            self.newpath = self.root + '/' + parts[1 if len(parts) > 1 else 0]
             if not os.path.exists(self.newpath):
                 os.makedirs(self.newpath)
             stl_files = msn_instance.dataset_files(dataset, '.stl')

torch_geometric/llm/models/g_retriever.py

@@ -24,12 +24,14 @@ class GRetriever(torch.nn.Module):
 
     .. warning::
         This module has been tested with the following HuggingFace models
+        * :obj:`llm_to_use="meta-llama/Meta-Llama-3.1-8B-Instruct"`
+        * :obj:`llm_to_use="Qwen/Qwen3-0.6B"`
 
-        * :obj:`llm_to_use="meta-llama/Llama-2-7b-chat-hf"`
-        * :obj:`llm_to_use="google/gemma-7b"`
 
-        and may not work with other models. See other models at `HuggingFace
-        Models <https://huggingface.co/models>`_ and let us know if you
+        This module should work with any HuggingFace model.
+        See other models at `HuggingFace
+        Models <https://huggingface.co/models>`_
+        and let us know if you
         encounter any issues.
 
     .. note::

torch_geometric/profile/utils.py

@@ -123,10 +123,9 @@ def get_gpu_memory_from_nvidia_smi(  # pragma: no cover
         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:
+            try:
+                mem_list.append(int(line.split()[0]))
+            except (TypeError, ValueError):
                 mem_list.append(None)
         return mem_list