pyg-nightly 2.7.0.dev20250905__py3-none-any.whl → 2.7.0.dev20250907__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (34) hide show
  1. {pyg_nightly-2.7.0.dev20250905.dist-info → pyg_nightly-2.7.0.dev20250907.dist-info}/METADATA +2 -1
  2. {pyg_nightly-2.7.0.dev20250905.dist-info → pyg_nightly-2.7.0.dev20250907.dist-info}/RECORD +32 -25
  3. torch_geometric/__init__.py +1 -1
  4. torch_geometric/data/__init__.py +0 -5
  5. torch_geometric/data/lightning/datamodule.py +2 -2
  6. torch_geometric/datasets/molecule_gpt_dataset.py +1 -1
  7. torch_geometric/datasets/web_qsp_dataset.py +262 -210
  8. torch_geometric/graphgym/imports.py +2 -2
  9. torch_geometric/llm/__init__.py +9 -0
  10. torch_geometric/{data → llm}/large_graph_indexer.py +124 -61
  11. torch_geometric/llm/models/__init__.py +23 -0
  12. torch_geometric/{nn → llm}/models/g_retriever.py +68 -49
  13. torch_geometric/{nn → llm}/models/git_mol.py +1 -1
  14. torch_geometric/{nn/nlp → llm/models}/llm.py +167 -33
  15. torch_geometric/llm/models/llm_judge.py +158 -0
  16. torch_geometric/{nn → llm}/models/molecule_gpt.py +1 -1
  17. torch_geometric/{nn/nlp → llm/models}/sentence_transformer.py +42 -8
  18. torch_geometric/llm/models/txt2kg.py +353 -0
  19. torch_geometric/llm/rag_loader.py +154 -0
  20. torch_geometric/llm/utils/backend_utils.py +442 -0
  21. torch_geometric/llm/utils/feature_store.py +169 -0
  22. torch_geometric/llm/utils/graph_store.py +199 -0
  23. torch_geometric/llm/utils/vectorrag.py +124 -0
  24. torch_geometric/loader/__init__.py +0 -4
  25. torch_geometric/nn/__init__.py +0 -1
  26. torch_geometric/nn/models/__init__.py +0 -10
  27. torch_geometric/nn/models/sgformer.py +2 -0
  28. torch_geometric/loader/rag_loader.py +0 -107
  29. torch_geometric/nn/nlp/__init__.py +0 -9
  30. {pyg_nightly-2.7.0.dev20250905.dist-info → pyg_nightly-2.7.0.dev20250907.dist-info}/WHEEL +0 -0
  31. {pyg_nightly-2.7.0.dev20250905.dist-info → pyg_nightly-2.7.0.dev20250907.dist-info}/licenses/LICENSE +0 -0
  32. /torch_geometric/{nn → llm}/models/glem.py +0 -0
  33. /torch_geometric/{nn → llm}/models/protein_mpnn.py +0 -0
  34. /torch_geometric/{nn/nlp → llm/models}/vision_transformer.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: pyg-nightly
3
- Version: 2.7.0.dev20250905
3
+ Version: 2.7.0.dev20250907
4
4
  Summary: Graph Neural Network Library for PyTorch
5
5
  Keywords: deep-learning,pytorch,geometric-deep-learning,graph-neural-networks,graph-convolutional-networks
6
6
  Author-email: Matthias Fey <matthias@pyg.org>
@@ -67,6 +67,7 @@ Requires-Dist: transformers ; extra == "rag"
67
67
  Requires-Dist: pandas ; extra == "rag"
68
68
  Requires-Dist: sentencepiece ; extra == "rag"
69
69
  Requires-Dist: accelerate ; extra == "rag"
70
+ Requires-Dist: torchmetrics ; extra == "rag"
70
71
  Requires-Dist: onnx ; extra == "test"
71
72
  Requires-Dist: onnxruntime ; extra == "test"
72
73
  Requires-Dist: onnxscript ; extra == "test"
@@ -1,4 +1,4 @@
1
- torch_geometric/__init__.py,sha256=JaI2udwlYGKK9_OGrMEQn05CBfOSEPFFsk9OtBnPN2c,2292
1
+ torch_geometric/__init__.py,sha256=etzB00fNBBr-taLZC_25zhXIa7jJAdOXPUbgKqNSgGQ,2292
2
2
  torch_geometric/_compile.py,sha256=9yqMTBKatZPr40WavJz9FjNi7pQj8YZAZOyZmmRGXgc,1351
3
3
  torch_geometric/_onnx.py,sha256=ODB_8cwFUiwBUjngXn6-K5HHb7IDul7DDXuuGX7vj_0,8178
4
4
  torch_geometric/backend.py,sha256=lVaf7aLoVaB3M-UcByUJ1G4T4FOK6LXAg0CF4W3E8jo,1575
@@ -30,7 +30,7 @@ torch_geometric/contrib/nn/conv/__init__.py,sha256=lrGnWsEiJf5zsBRmshGZZFN_uYR2e
30
30
  torch_geometric/contrib/nn/models/__init__.py,sha256=3ia5cX-TPhouLl6jn_HA-Rd2LaaQvFgy5CjRk0ovKRU,113
31
31
  torch_geometric/contrib/nn/models/rbcd_attack.py,sha256=qcyxBxAbx8LKzpp3RoJQ0cxl9aB2onsWT4oY1fsM7us,33280
32
32
  torch_geometric/contrib/transforms/__init__.py,sha256=lrGnWsEiJf5zsBRmshGZZFN_uYR2ezDjbj9n9nCpvtk,23
33
- torch_geometric/data/__init__.py,sha256=D6Iz5A9vEb_2rpf96Zn7uM-lchZ3WpW8X7WdAD1yxKw,4565
33
+ torch_geometric/data/__init__.py,sha256=OLkV82AGm6xMSynT_DHfRE6_INfPxLx4BQnY0-WVn54,4323
34
34
  torch_geometric/data/batch.py,sha256=8X8CN4_1rjrh48R3R2--mZUgfsO7Po9JP-H6SbrBiBA,8740
35
35
  torch_geometric/data/collate.py,sha256=tOUvttXoEo-bOvJx_qMivJq2JqOsB9iDdjovtiyys4o,12644
36
36
  torch_geometric/data/data.py,sha256=-E6el1knNgSJyapV8KUk2aRRHOfvwEvjUFfe_BapLfc,47490
@@ -44,7 +44,6 @@ torch_geometric/data/graph_store.py,sha256=dSMCcMYlka2elfw-Rof-lG_iGQv6NHX98uPEV
44
44
  torch_geometric/data/hetero_data.py,sha256=i4_Wt8MFyEG1ZhsU_zg7IKqA2ZatrYyjtYs_R_Qrf3U,53353
45
45
  torch_geometric/data/hypergraph_data.py,sha256=LfriiuJRx9ZrrSGj_fO5NUsh4kvyXJuRdCOqsWo__vc,8307
46
46
  torch_geometric/data/in_memory_dataset.py,sha256=ilFxjF4pvBILsS4wOqocwRBc2j6toI2S_KMFF19KB1w,13413
47
- torch_geometric/data/large_graph_indexer.py,sha256=myXTXhbRHQPxEOHNHPeNHB_pBzXCIQBr1KQt9WwBoi8,25468
48
47
  torch_geometric/data/makedirs.py,sha256=6uOv4y34i947cm4rv7Aj2_YZBq-EOsyPKnlGA188YSw,463
49
48
  torch_geometric/data/on_disk_dataset.py,sha256=77om-e6kzcpBb77kf7um1xY8-yHmQaao_6R7I-3NwHk,6629
50
49
  torch_geometric/data/remote_backend_utils.py,sha256=Rzpq1PczXuHhUscrFtIAL6dua6pMehSJlXG7yEsrrrg,4503
@@ -54,7 +53,7 @@ torch_geometric/data/summary.py,sha256=o9rmAIkPEAMpsjd4UhoIWtpGsCd0oHpBS99_XkATb
54
53
  torch_geometric/data/temporal.py,sha256=WOJ6gFrTLikaLhUvotyUF5ql14FkE5Ox3hNkdSp6ZGg,10018
55
54
  torch_geometric/data/view.py,sha256=XjkVSc-UWZFCT4DlXLShZtO8duhFQkS9gq88zZXANsk,1089
56
55
  torch_geometric/data/lightning/__init__.py,sha256=w3En1tJfy3kSqe1MycpOyZpHFO3fxBCgNCUOznPA3YU,178
57
- torch_geometric/data/lightning/datamodule.py,sha256=jDv9ibLQV_FyPeq8ncq77oOU8qy-STCf-aaYd0R8JE8,29545
56
+ torch_geometric/data/lightning/datamodule.py,sha256=uC7ct5cIUzLsKH-Lu1O2ly1NIJiwXYYLP-pTkoVxUS4,29545
58
57
  torch_geometric/datasets/__init__.py,sha256=rgfUmjd9U3o8renKVl81Brscx4LOtwWmt6qAoaG41C4,6417
59
58
  torch_geometric/datasets/actor.py,sha256=oUxgJIX8bi5hJr1etWNYIFyVQNDDXi1nyVpHGGMEAGQ,4304
60
59
  torch_geometric/datasets/airfrans.py,sha256=Pc9C7IuEKkKzko_RmFPQ5gzOAGJ3132DoZZ4HaePBT8,5440
@@ -119,7 +118,7 @@ torch_geometric/datasets/medshapenet.py,sha256=eCBCXKpueweCwDSf_Q4_MwVA3IbJd04FS
119
118
  torch_geometric/datasets/mixhop_synthetic_dataset.py,sha256=4NNvTHUvvV6pcqQCyVDS5XhppXUeF2H9GTfFoc49eyU,3951
120
119
  torch_geometric/datasets/mnist_superpixels.py,sha256=o2ArbZ0_OE0u8VCaHmWwvngESlOFr9oM9dSEP_tjAS4,3340
121
120
  torch_geometric/datasets/modelnet.py,sha256=rqR-e75lC8PS_IX7VlNbo2Az9IWfqMNvDp8rmQCp-LE,5357
122
- torch_geometric/datasets/molecule_gpt_dataset.py,sha256=S7Zc-OJ4kns1L8vFJ--3QiMF6xKbmoccGZUgQpVBe1o,19087
121
+ torch_geometric/datasets/molecule_gpt_dataset.py,sha256=d6V8_Qy1Y2nt9hFhn7Re1omFw5Tf_uhv13QM0Vg76eg,19091
123
122
  torch_geometric/datasets/molecule_net.py,sha256=pMzaJzd-LbBncZ0VoC87HfA8d1F4NwCWTb5YKvLM890,7404
124
123
  torch_geometric/datasets/movie_lens.py,sha256=M4Bu0Xus8IkW8GYzjxPxSdPXNbcCCx9cu6cncxBvLx8,4033
125
124
  torch_geometric/datasets/movie_lens_100k.py,sha256=eTpBAteM3jqTEtiwLxmhVj4r8JvftvPx8Hvs-3ZIHlU,6057
@@ -159,7 +158,7 @@ torch_geometric/datasets/tosca.py,sha256=nUSF8NQT1GlkwWQLshjWmr8xORsvRHzzIqhUyDC
159
158
  torch_geometric/datasets/tu_dataset.py,sha256=14OSaXBgVwT1dX2h1wZ3xVIwoo0GQBEfR3yWh6Q0VF0,7847
160
159
  torch_geometric/datasets/twitch.py,sha256=qfEerf-Uaojx2ZvegENowdG4E7RoUT_HUO9xtULadvo,3658
161
160
  torch_geometric/datasets/upfd.py,sha256=crqO8uQNz1wC1JOn4prSs8iOGv9LuLK3dZf_KUV9tUE,7010
162
- torch_geometric/datasets/web_qsp_dataset.py,sha256=7ygTK2LGRvJtnlyM3rl-9GV3VpRQWQY_ettKoJRnXz4,11120
161
+ torch_geometric/datasets/web_qsp_dataset.py,sha256=2MyPFphyLN3DavjzOPuW2s-4v6yWIm9GReDeI6K1-xY,14400
163
162
  torch_geometric/datasets/webkb.py,sha256=beC1kWeW7cIjYwWyaINQSk-3lmVR85Lus7cKZniHp8Y,4879
164
163
  torch_geometric/datasets/wikics.py,sha256=1aELKdZKqqQidkwRSNCJUTtACCbrlZChOuYxakD3fRc,3909
165
164
  torch_geometric/datasets/wikidata.py,sha256=9mYShF_HlpTmcdLpiaP_tYJ9eQtUOu5vRPvohN6RXqI,4979
@@ -217,7 +216,7 @@ torch_geometric/graphgym/benchmark.py,sha256=52Y8fyMQ5Q0hS0kowBlVfnnk-pkLNQLFc8l
217
216
  torch_geometric/graphgym/checkpoint.py,sha256=v07NKxmXLn0Rqy87t0jHZtyJojRSMWdcwRRdZyWCY18,2408
218
217
  torch_geometric/graphgym/cmd_args.py,sha256=l4v_o21YYQ1uaWcKpeMz-D5_CfHKfNDQQbqatomnDgw,738
219
218
  torch_geometric/graphgym/config.py,sha256=e0TLYKlC_O3VANaYWRfKbjttFb-IhmJayOol5OiPs9g,17210
220
- torch_geometric/graphgym/imports.py,sha256=ymrPip1soU36szGl0YS0zs00Qf4_dG7exuCRfk8upUM,615
219
+ torch_geometric/graphgym/imports.py,sha256=UqdzEWDvl_vlPJYWxfOQCJiQ57tpI6uLhXWe37sRLmY,615
221
220
  torch_geometric/graphgym/init.py,sha256=qRlA1BmbtP-ve8XsD4HKpczWRnBMddtI2DdE1pHfJFQ,521
222
221
  torch_geometric/graphgym/loader.py,sha256=CEJv8Hw4kwAWr6A7f9GCoU0UIb4K4zWYDx29L7xfiJ8,11605
223
222
  torch_geometric/graphgym/logger.py,sha256=6vqMLOvxPPQcCA2o3W-jaX2PZwZt_rR66NktBwgH48Q,11272
@@ -268,7 +267,25 @@ torch_geometric/io/ply.py,sha256=NdeTtr79vJ1HS37ZV2N61EUmA5NGJd2I6cUj1Pg7Ypg,489
268
267
  torch_geometric/io/sdf.py,sha256=H2PC6dSW9Kncc1ulb0UN0JnTRT93NY2fY8lf6K4hb50,1165
269
268
  torch_geometric/io/tu.py,sha256=-v5Ago7DfmGTRBtB5RZFvmv4XpLnKKnk-NOnxlHtB_c,4881
270
269
  torch_geometric/io/txt_array.py,sha256=LDeX2qtlNKW-kVe-wpnskMwAdXQp1jVCGQnrJce7Smg,910
271
- torch_geometric/loader/__init__.py,sha256=DJrdCD1A5PuBYRSgxFbZU9GTBStNuKngqkUV1oEfQQ4,1971
270
+ torch_geometric/llm/__init__.py,sha256=ikGCkUbn_gjlq6GPSDu43sErVN8syBetNNasqtn9X74,217
271
+ torch_geometric/llm/large_graph_indexer.py,sha256=ip5wU7Rj4S1tFc53AqAtvyvGqeVC7AJrVMEbRTHyz54,28429
272
+ torch_geometric/llm/rag_loader.py,sha256=lsUt5kvHGbQzxOuWubBVB0RkMcTi6b4RNRuvsArw3Ys,5641
273
+ torch_geometric/llm/models/__init__.py,sha256=nzO63kzvbWNRicrxvmRl0m622WEAZ6_x8XI_hCocTH4,537
274
+ torch_geometric/llm/models/g_retriever.py,sha256=SeW0rlrlzMhN3aVslhq_GUyUgS4sVw_nMAT5YiXzZd8,9072
275
+ torch_geometric/llm/models/git_mol.py,sha256=m1YJb6Xb2i6j9wEqHzqE4YBWMr9i1CJpMp6T-E24fsA,12680
276
+ torch_geometric/llm/models/glem.py,sha256=GlL_I63g-_5eTycSGRj720YntldQ-CQ351RaDPc6XAU,16674
277
+ torch_geometric/llm/models/llm.py,sha256=LKYdUSw2PSWYEN13DYYaJt0dHRQTptX6mM1rAh7-J6c,17764
278
+ torch_geometric/llm/models/llm_judge.py,sha256=qhc8hmIPNhcfLVRyBVk7jQW7ncoIb9QYw7rcsGAIpyg,6457
279
+ torch_geometric/llm/models/molecule_gpt.py,sha256=RWoP4RMsoRzZtuedPCLNCfooqibCqxkuAhH-pyek9No,7641
280
+ torch_geometric/llm/models/protein_mpnn.py,sha256=SwTgafSbI2KJ-yqzn0trZtVWLmfo0_kPEaWSNJUCt70,12266
281
+ torch_geometric/llm/models/sentence_transformer.py,sha256=vubAluqfxVKevuPpxl0c43ps_f3N0ZMi1_cRIV9EwuA,6076
282
+ torch_geometric/llm/models/txt2kg.py,sha256=CjWXCa_WrqIlBfOSPMAK-mnaX19dvH_YhXMWSuLkk4o,14074
283
+ torch_geometric/llm/models/vision_transformer.py,sha256=diVBefjIynzYs8WBlcpTeSVnw1PUecHY--B9Yd-W2hA,863
284
+ torch_geometric/llm/utils/backend_utils.py,sha256=AcvZ8ym3UKCWwWoMl8sNRnj4FSl0_srHw9yxZAKiyeU,15840
285
+ torch_geometric/llm/utils/feature_store.py,sha256=CMmDveFxLGfCUdyo0c0sygLLplZlIj-Fwxuk1kKxpQs,5885
286
+ torch_geometric/llm/utils/graph_store.py,sha256=_Hh0aGnokUn0zvOC80xUfT4TtX_7G4KIDoEBkNXkgHY,7103
287
+ torch_geometric/llm/utils/vectorrag.py,sha256=m51drMNXsGQTN6qkbR8QiHb8jvcrBlZDHzEjsEmtnes,4753
288
+ torch_geometric/loader/__init__.py,sha256=w9LSTbyrLRkyrLXi_10d80csWgfKOKDRQDJXRdcfD0M,1835
272
289
  torch_geometric/loader/base.py,sha256=ataIwNEYL0px3CN3LJEgXIVTRylDHB6-yBFXXuX2JN0,1615
273
290
  torch_geometric/loader/cache.py,sha256=S65heO3YTyUPbttqizCNtKPHIoAw5iHRpbvw6KlXmok,2106
274
291
  torch_geometric/loader/cluster.py,sha256=CbZUy739vzMqOKgof2N73uc-Br4Daw56G3XMzptLUT8,13469
@@ -287,7 +304,6 @@ torch_geometric/loader/neighbor_loader.py,sha256=vnLn_RhBKTux5h8pi0vzj0d7JPoOpLA
287
304
  torch_geometric/loader/neighbor_sampler.py,sha256=mraVFXIIGctYot4Xr2VOAhCKAOQyW2gP9KROf7g6tcc,8497
288
305
  torch_geometric/loader/node_loader.py,sha256=g_kV5N0tO6eMSFPc5fdbzfHr4COAeKVJi7FEq52f4zc,11848
289
306
  torch_geometric/loader/prefetch.py,sha256=z30TIcu3_6ZubllUOwNLunlq4RyQdFj36vPE5QE_NkM,3232
290
- torch_geometric/loader/rag_loader.py,sha256=Q2YvFTe4qoThUvsmCQa-4icRvtjBwbTOkWOAQDOhhG4,4642
291
307
  torch_geometric/loader/random_node_loader.py,sha256=rCmRXYv70SPxBo-Oh049eFEWEZDV7FmlRPzmjcoirXQ,2196
292
308
  torch_geometric/loader/shadow.py,sha256=_hCspYf9SlJYX0lqEjxFec9e9t1iMScNThOoWR1wQGM,4173
293
309
  torch_geometric/loader/temporal_dataloader.py,sha256=Z7L_rYdl6SYBQXAgtr18FVcmfMH9kP1fBWrc2W63g2c,2250
@@ -295,7 +311,7 @@ torch_geometric/loader/utils.py,sha256=3hzKzIgB52QIZu7Jdn4JeXZaegIJinIQfIUP9DrUW
295
311
  torch_geometric/loader/zip_loader.py,sha256=3lt10fD15Rxm1WhWzypswGzCEwUz4h8OLCD1nE15yNg,3843
296
312
  torch_geometric/metrics/__init__.py,sha256=3krvDobW6vV5yHTjq2S2pmOXxNfysNG26muq7z48e94,699
297
313
  torch_geometric/metrics/link_pred.py,sha256=1_hE3KiRqAdZLI6QuUbjgyFC__mTyFu_RimM3bD8wRw,31678
298
- torch_geometric/nn/__init__.py,sha256=kQHHHUxFDht2ztD-XFQuv98TvC8MdodaFsIjAvltJBw,874
314
+ torch_geometric/nn/__init__.py,sha256=RrWRzEoqtR3lsO2lAzYXboLPb3uYEX2z3tLxiBIVWjc,847
299
315
  torch_geometric/nn/data_parallel.py,sha256=YiybTWoSFyfSzlXAamZ_-y1f7B6tvDEFHOuy_AyJz9Q,4761
300
316
  torch_geometric/nn/encoding.py,sha256=3DCOCO-XFt-lMb97sHWGN-4KeGUFY5lVo9P00SzrCNk,3559
301
317
  torch_geometric/nn/fx.py,sha256=PDtaHJAgodh4xf8FNl4fVxPGZJDbRaq3Q9z8qb1DNNI,16063
@@ -432,7 +448,7 @@ torch_geometric/nn/kge/distmult.py,sha256=dGQ0bVzjreZgFN1lXE23_IIidsiOq7ehPrMb-N
432
448
  torch_geometric/nn/kge/loader.py,sha256=5Uc1j3OUMQnBYSHDqL7pLCty1siFLzoPkztigYO2zP8,771
433
449
  torch_geometric/nn/kge/rotate.py,sha256=XLuO1AbyTt5cJxr97ZzoyAyIEsHKesgW5TvDmnGJAao,3208
434
450
  torch_geometric/nn/kge/transe.py,sha256=jlejq5BLMm-sb1wWcLDp7pZqCdelWBgjDIC8ctbjSdU,3088
435
- torch_geometric/nn/models/__init__.py,sha256=2INurndMehxviZcdD7REV7qnIaLJ9AXdnNvcwqa00HM,2496
451
+ torch_geometric/nn/models/__init__.py,sha256=7eRlAR93pltDfLEcsTJaaK67mVFezAIjg0VIbUEeTSw,2251
436
452
  torch_geometric/nn/models/attentive_fp.py,sha256=1z3iTV2O5W9tqHFAdno8FeBFeXmuG-TDZk4lwwVh3Ac,6634
437
453
  torch_geometric/nn/models/attract_repel.py,sha256=h9OyogT0NY0xiT0DkpJHMxH6ZUmo8R-CmwZdKEwq8Ek,5277
438
454
  torch_geometric/nn/models/autoencoder.py,sha256=nGje-zty78Y3hxOJ9o0_6QziJjOvBlknk6z0_fDQwQU,10770
@@ -443,9 +459,6 @@ torch_geometric/nn/models/deep_graph_infomax.py,sha256=yXSZ4mCrq4Dcvl1muzkxEWH4L
443
459
  torch_geometric/nn/models/deepgcn.py,sha256=tIgT03cj8MghYlxEozpoGvGG_CwpJrGDxv1Z0CVIUts,4339
444
460
  torch_geometric/nn/models/dimenet.py,sha256=O2rqEx5HWs_lMwRD8eq6WMkbqJaCLL5zgWUJ27qh9Bc,36188
445
461
  torch_geometric/nn/models/dimenet_utils.py,sha256=Eyn_EiJqwKvuYj6BtRpSxrzMG3v4Gk98X9MxZ7uvwm4,5069
446
- torch_geometric/nn/models/g_retriever.py,sha256=tVibbqM_r-1LnA3R3oVyzp0bpuN3qPoYqcU6LZ8dYEk,8260
447
- torch_geometric/nn/models/git_mol.py,sha256=Wc6Hx6RDDR7sDWRWHfA5eK9e9gFsrTZ9OLmpMfoj3pE,12676
448
- torch_geometric/nn/models/glem.py,sha256=GlL_I63g-_5eTycSGRj720YntldQ-CQ351RaDPc6XAU,16674
449
462
  torch_geometric/nn/models/gnnff.py,sha256=15dkiLgy0LmH1hnUrpeoHioIp4BPTfjpVATpnGRt9E0,7860
450
463
  torch_geometric/nn/models/gpse.py,sha256=acEAeeicLgzKRL54WhvIFxjA5XViHgXgMEH-NgbMdqI,41971
451
464
  torch_geometric/nn/models/graph_mixer.py,sha256=mthMeCOikR8gseEsu4oJ3Cd9C35zHSv1p32ROwnG-6s,9246
@@ -459,24 +472,18 @@ torch_geometric/nn/models/mask_label.py,sha256=B2HcL6ZkaUEo3a8nebZoUqEIfDEfcIGOV
459
472
  torch_geometric/nn/models/meta.py,sha256=lQWovjdQgTGT_rDAm6L186ObINeQCD9tLBz8xenmrF0,6540
460
473
  torch_geometric/nn/models/metapath2vec.py,sha256=jwpEwVUnt9uWcriVfyX1BVCzXOqKpnDVsPQSxJAGoBQ,10788
461
474
  torch_geometric/nn/models/mlp.py,sha256=K-WOLQO-DBtaDdMw7zoEbPgItDQg6DCrsd89B6-HvCE,10352
462
- torch_geometric/nn/models/molecule_gpt.py,sha256=k-XULH6jaurj-R2EE4sIWTkqlNqa3CzWxfQgfFa-G8s,7637
463
475
  torch_geometric/nn/models/neural_fingerprint.py,sha256=pTLJgU9Uh2Lnf9bggLj4cKI8YdEFcMF-9MALuubqbuQ,2378
464
476
  torch_geometric/nn/models/node2vec.py,sha256=81Ku4Rp4IwLEAy06KEgJ2fYtXXVL_uv_Hb8lBr6YXrE,7664
465
477
  torch_geometric/nn/models/pmlp.py,sha256=dcAASVSyQMMhItSfEJWPeAFh0R3tNCwAHwdrShwQ8o4,3538
466
478
  torch_geometric/nn/models/polynormer.py,sha256=JgUngkF18sgepAAJTO7js9RISmYLWiO04-JEeV4J__8,7641
467
- torch_geometric/nn/models/protein_mpnn.py,sha256=SwTgafSbI2KJ-yqzn0trZtVWLmfo0_kPEaWSNJUCt70,12266
468
479
  torch_geometric/nn/models/re_net.py,sha256=pz66q5b5BoGDNVQvpEGS2RGoeKvpjkYAv9r3WAuvITk,8986
469
480
  torch_geometric/nn/models/rect.py,sha256=2F3XyyvHTAEuqfJpiNB5M8pSGy738LhPiom5I-SDWqM,2808
470
481
  torch_geometric/nn/models/rev_gnn.py,sha256=bkKBAd_vXZ3UDMTJgdVObqleYHOTVsVcisftK7XoDlo,11797
471
482
  torch_geometric/nn/models/schnet.py,sha256=0aaHrVtxApdvn3RHCGLQJW1MbIb--CSYUrx9O3hDOZM,16656
472
- torch_geometric/nn/models/sgformer.py,sha256=3NDzkEVRtM1QmeJsXSq7FBhGGchyUvyX1SDPKYg9F70,6875
483
+ torch_geometric/nn/models/sgformer.py,sha256=Jko3gQtiSanARhZSfuujpbSS1vvV_UMVq902SEgbfSI,6917
473
484
  torch_geometric/nn/models/signed_gcn.py,sha256=HEKaXZIWoDnsBRxIytviTpwsjQIFKl44c9glNUpwhlM,9841
474
485
  torch_geometric/nn/models/tgn.py,sha256=kEGdfLJybkbMT4UMoAh2nCzfX3_nDjfm1cicuPHEwAM,11878
475
486
  torch_geometric/nn/models/visnet.py,sha256=-ABewG-3BzChboHAvoiO3bbi3d_46x8egyvTwxBD5WE,43194
476
- torch_geometric/nn/nlp/__init__.py,sha256=q6CPUiJHcc9bXw90lyj-ID4F3kfW8uPM-SOxW9uCMHs,213
477
- torch_geometric/nn/nlp/llm.py,sha256=DAv9jOZKXKQNVU2pNMyS1q8gVUtlin_unc6FjLhOYto,12267
478
- torch_geometric/nn/nlp/sentence_transformer.py,sha256=q5M7SGtrUzoSiNhKCGFb7JatWiukdhNF6zdq2yiqxwE,4475
479
- torch_geometric/nn/nlp/vision_transformer.py,sha256=diVBefjIynzYs8WBlcpTeSVnw1PUecHY--B9Yd-W2hA,863
480
487
  torch_geometric/nn/norm/__init__.py,sha256=u2qIDrkbeuObGVXSAIftAlvSd6ouGTtxznCfD-59UiA,669
481
488
  torch_geometric/nn/norm/batch_norm.py,sha256=fzUNmpdCUsMnNcso3PKDUdWc0UQvziK80-w0ZC0Vb8U,8706
482
489
  torch_geometric/nn/norm/diff_group_norm.py,sha256=mT0gM5a8txcAFNwZGKFu12qnNF5Pn95zrMx-RisRsh4,4938
@@ -646,7 +653,7 @@ torch_geometric/utils/undirected.py,sha256=H_nfpI0_WluOG6VfjPyldvcjL4w5USAKWu2x5
646
653
  torch_geometric/visualization/__init__.py,sha256=b-HnVesXjyJ_L1N-DnjiRiRVf7lhwKaBQF_2i5YMVSU,208
647
654
  torch_geometric/visualization/graph.py,sha256=mfZHXYfiU-CWMtfawYc80IxVwVmtK9hbIkSKhM_j7oI,14311
648
655
  torch_geometric/visualization/influence.py,sha256=CWMvuNA_Nf1sfbJmQgn58yS4OFpeKXeZPe7kEuvkUBw,477
649
- pyg_nightly-2.7.0.dev20250905.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
650
- pyg_nightly-2.7.0.dev20250905.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
651
- pyg_nightly-2.7.0.dev20250905.dist-info/METADATA,sha256=7zFs8SgMgRdkwK679QLO1W08UGscHMakHb-3NXsiEok,64100
652
- pyg_nightly-2.7.0.dev20250905.dist-info/RECORD,,
656
+ pyg_nightly-2.7.0.dev20250907.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
657
+ pyg_nightly-2.7.0.dev20250907.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
658
+ pyg_nightly-2.7.0.dev20250907.dist-info/METADATA,sha256=_vdXep8HARAY4LRARD8f0bRvEBZ2Zk-Os89Pz7Zo6qM,64145
659
+ pyg_nightly-2.7.0.dev20250907.dist-info/RECORD,,
@@ -31,7 +31,7 @@ from .lazy_loader import LazyLoader
31
31
  contrib = LazyLoader('contrib', globals(), 'torch_geometric.contrib')
32
32
  graphgym = LazyLoader('graphgym', globals(), 'torch_geometric.graphgym')
33
33
 
34
- __version__ = '2.7.0.dev20250905'
34
+ __version__ = '2.7.0.dev20250907'
35
35
 
36
36
  __all__ = [
37
37
  'Index',
@@ -16,7 +16,6 @@ from .on_disk_dataset import OnDiskDataset
16
16
  from .makedirs import makedirs
17
17
  from .download import download_url, download_google_url
18
18
  from .extract import extract_tar, extract_zip, extract_bz2, extract_gz
19
- from .large_graph_indexer import LargeGraphIndexer, TripletLike, get_features_for_triplets, get_features_for_triplets_groups
20
19
 
21
20
  from torch_geometric.lazy_loader import LazyLoader
22
21
 
@@ -28,8 +27,6 @@ data_classes = [
28
27
  'Dataset',
29
28
  'InMemoryDataset',
30
29
  'OnDiskDataset',
31
- 'LargeGraphIndexer',
32
- 'TripletLike',
33
30
  ]
34
31
 
35
32
  remote_backend_classes = [
@@ -53,8 +50,6 @@ helper_functions = [
53
50
  'extract_zip',
54
51
  'extract_bz2',
55
52
  'extract_gz',
56
- 'get_features_for_triplets',
57
- "get_features_for_triplets_groups",
58
53
  ]
59
54
 
60
55
  __all__ = data_classes + remote_backend_classes + helper_functions
@@ -11,11 +11,11 @@ from torch_geometric.sampler import BaseSampler, NeighborSampler
11
11
  from torch_geometric.typing import InputEdges, InputNodes, OptTensor
12
12
 
13
13
  try:
14
- from pytorch_lightning import LightningDataModule as _LightningDataModule
14
+ from lightning.pytorch import LightningDataModule as _LightningDataModule
15
15
  _pl_is_available = True
16
16
  except ImportError:
17
17
  try:
18
- from lightning.pytorch import \
18
+ from pytorch_lightning import \
19
19
  LightningDataModule as _LightningDataModule
20
20
  _pl_is_available = True
21
21
  except ImportError:
@@ -14,7 +14,7 @@ from tqdm import tqdm
14
14
 
15
15
  from torch_geometric.data import Data, InMemoryDataset, download_url
16
16
  from torch_geometric.io import fs
17
- from torch_geometric.nn.nlp import LLM
17
+ from torch_geometric.llm.models import LLM
18
18
  from torch_geometric.utils import one_hot
19
19
 
20
20