pyg-nightly 2.7.0.dev20251031__py3-none-any.whl → 2.7.0.dev20251101__py3-none-any.whl

{pyg_nightly-2.7.0.dev20251031.dist-info → pyg_nightly-2.7.0.dev20251101.dist-info}/METADATA

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

@@ -1,4 +1,4 @@
-torch_geometric/__init__.py,sha256=QiSp5I-qmE7LznK_oirFMHoLFkb4ZSMY7aRGvSS87eE,2292
+torch_geometric/__init__.py,sha256=RqO8G1pJjRlPfkJmybH_Jth5W7eYVFllFtiTOgZSDBk,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
@@ -151,7 +151,7 @@ torch_geometric/datasets/shapenet.py,sha256=tn3HiQQAr6lxHrqxfOVaAtl40guwFYTXWCbS
 torch_geometric/datasets/shrec2016.py,sha256=cTLhctbqE0EUEvKddJFhPzDb1oLKXOth4O_WzsWtyMk,6323
 torch_geometric/datasets/snap_dataset.py,sha256=deJvB6cpIQ3bu_pcWoqgEo1-Kl_NcFi7ZSUci645X0U,9481
 torch_geometric/datasets/suite_sparse.py,sha256=eqjH4vAUq872qdk3YdLkZSwlu6r7HHpTgK0vEVGmY1s,3278
-torch_geometric/datasets/tag_dataset.py,sha256=jslijGCh37ip2YkrQLyvbk-1QRJ3yqFpmzuQSxckXrE,19402
+torch_geometric/datasets/tag_dataset.py,sha256=TwZMDxnFGjl4vPosZIp1mmqMXjxcMt3WcgeiTeKMjuU,19788
 torch_geometric/datasets/taobao.py,sha256=CUcZpbWsNTasevflO8zqP0YvENy89P7wpKS4MHaDJ6Q,4170
 torch_geometric/datasets/teeth3ds.py,sha256=hZvhcq9lsQENNFr5hk50w2T3CgxE_tlnQfrCgN6uIDQ,9919
 torch_geometric/datasets/tosca.py,sha256=nUSF8NQT1GlkwWQLshjWmr8xORsvRHzzIqhUyDCvABc,4632
@@ -319,7 +319,7 @@ torch_geometric/nn/fx.py,sha256=PDtaHJAgodh4xf8FNl4fVxPGZJDbRaq3Q9z8qb1DNNI,1606
 torch_geometric/nn/glob.py,sha256=MdHjcUlHmFmTevzwND1_x7dXXJPzIDTBJRGOrGdZ8dQ,1088
 torch_geometric/nn/inits.py,sha256=_8FqacCLPz5Ft2zB5s6dtKGTKWtfrLyCLLuv1QvyKjk,2457
 torch_geometric/nn/lr_scheduler.py,sha256=_FWdIgGPDSZCK1TZFWHSP5RfpY83Kyhlz7Ja6YHPQVo,8937
-torch_geometric/nn/model_hub.py,sha256=2x8uN80BwW2pdBGtU4LNF-8pUtGZMxr0btQCVSgeKwA,9550
+torch_geometric/nn/model_hub.py,sha256=RNzpkm1D2pJTO3hwicoxxEcummd4ukTTw5anxqVT2mc,9561
 torch_geometric/nn/module_dict.py,sha256=c6ThkEfju01ZyhERA8rZgr9oawVfwqOBMrdG3bzSsro,2377
 torch_geometric/nn/parameter_dict.py,sha256=RJnid_I8jXTjIoDKVr0Ej2E2XKOLxH_lXxLeEHB_wUg,2412
 torch_geometric/nn/reshape.py,sha256=WJvc_oPPcfsFqN97i4oJRMor17ar5IPeIu8BevKIyIk,426
@@ -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.dev20251031.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
-pyg_nightly-2.7.0.dev20251031.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-pyg_nightly-2.7.0.dev20251031.dist-info/METADATA,sha256=6dbDyVNNRGJkd_eBVdtQMmXqW4-ms1yR-yZgBfl_484,63680
-pyg_nightly-2.7.0.dev20251031.dist-info/RECORD,,
+pyg_nightly-2.7.0.dev20251101.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
+pyg_nightly-2.7.0.dev20251101.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+pyg_nightly-2.7.0.dev20251101.dist-info/METADATA,sha256=FbjnIf8Wxp05LM1B6WTUoGMH01w2sx2vIfSeD4tDc6o,63680
+pyg_nightly-2.7.0.dev20251101.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.dev20251031'
+__version__ = '2.7.0.dev20251101'
 
 __all__ = [
     'Index',

torch_geometric/datasets/tag_dataset.py

@@ -137,10 +137,13 @@ class TAGDataset(InMemoryDataset):
         self.token_on_disk = token_on_disk
         self.tokenize_batch_size = tokenize_batch_size
         self._token = self.tokenize_graph(self.tokenize_batch_size)
-        self._llm_explanation_token = self.tokenize_graph(
-            self.tokenize_batch_size, text_type='llm_explanation')
-        self._all_token = self.tokenize_graph(self.tokenize_batch_size,
-                                              text_type='all')
+        self._llm_explanation_token: Dict[str, Tensor] = {}
+        self._all_token: Dict[str, Tensor] = {}
+        if self.name in self.llm_explanation_id:
+            self._llm_explanation_token = self.tokenize_graph(
+                self.tokenize_batch_size, text_type='llm_explanation')
+            self._all_token = self.tokenize_graph(self.tokenize_batch_size,
+                                                  text_type='all')
         self.__num_classes__ = dataset.num_classes
 
     @property
@@ -170,14 +173,16 @@ class TAGDataset(InMemoryDataset):
 
     @property
     def llm_explanation_token(self) -> Dict[str, Tensor]:
-        if self._llm_explanation_token is None:  # lazy load
+        if self._llm_explanation_token is None and \
+                self.name in self.llm_explanation_id:
             self._llm_explanation_token = self.tokenize_graph(
                 text_type='llm_explanation')
         return self._llm_explanation_token
 
     @property
     def all_token(self) -> Dict[str, Tensor]:
-        if self._all_token is None:  # lazy load
+        if self._all_token is None and \
+                self.name in self.llm_explanation_id:
             self._all_token = self.tokenize_graph(text_type='all')
         return self._all_token
 
@@ -230,13 +235,15 @@ class TAGDataset(InMemoryDataset):
                                             filename='node-text.csv.gz',
                                             log=True)
         self.text = list(read_csv(raw_text_path)['text'])
-        print('downloading llm explanations')
-        llm_explanation_path = download_google_url(
-            id=self.llm_explanation_id[self.name], folder=f'{self.root}/raw',
-            filename='node-gpt-response.csv.gz', log=True)
-        self.llm_explanation = list(read_csv(llm_explanation_path)['text'])
-        print('downloading llm predictions')
-        fs.cp(f'{self.llm_prediction_url}/{self.name}.csv', self.raw_dir)
+        if self.name in self.llm_explanation_id:
+            print('downloading llm explanations')
+            llm_explanation_path = download_google_url(
+                id=self.llm_explanation_id[self.name],
+                folder=f'{self.root}/raw', filename='node-gpt-response.csv.gz',
+                log=True)
+            self.llm_explanation = list(read_csv(llm_explanation_path)['text'])
+            print('downloading llm predictions')
+            fs.cp(f'{self.llm_prediction_url}/{self.name}.csv', self.raw_dir)
 
     def process(self) -> None:
         # process Title and Abstraction
@@ -276,6 +283,14 @@ class TAGDataset(InMemoryDataset):
             for i, pred in enumerate(preds):
                 pl[i][:len(pred)] = torch.tensor(
                     pred[:self.llm_prediction_topk], dtype=torch.long) + 1
+
+            if self.llm_explanation is None or pl is None:
+                raise ValueError(
+                    "The TAGDataset only have ogbn-arxiv LLM explanations"
+                    "and predictions in default. The llm explanation and"
+                    "prediction of each node is not specified.Please pass in"
+                    "'llm_explanation' and 'llm_prediction' when"
+                    "convert your dataset to Text Attribute Graph Dataset")
         elif self.name in self.llm_explanation_id:
             self.download()
         else:
@@ -283,13 +298,6 @@ class TAGDataset(InMemoryDataset):
                 'The dataset is not ogbn-arxiv,'
                 'please pass in your llm explanation list to `llm_explanation`'
                 'and llm prediction list to `llm_prediction`')
-        if self.llm_explanation is None or pl is None:
-            raise ValueError(
-                "The TAGDataset only have ogbn-arxiv LLM explanations"
-                "and predictions in default. The llm explanation and"
-                "prediction of each node is not specified."
-                "Please pass in 'llm_explanation' and 'llm_prediction' when"
-                "convert your dataset to Text Attribute Graph Dataset")
 
     def save_node_text(self, text: List[str]) -> None:
         node_text_path = osp.join(self.root, 'raw', 'node-text.csv.gz')

torch_geometric/nn/model_hub.py

@@ -144,10 +144,10 @@ class PyGModelHubMixin(ModelHubMixin):
         revision,
         cache_dir,
         force_download,
-        proxies,
-        resume_download,
         local_files_only,
         token,
+        proxies=None,
+        resume_download=False,
         dataset_name='',
         model_name='',
         map_location='cpu',