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.
- {pyg_nightly-2.7.0.dev20250905.dist-info → pyg_nightly-2.7.0.dev20250907.dist-info}/METADATA +2 -1
- {pyg_nightly-2.7.0.dev20250905.dist-info → pyg_nightly-2.7.0.dev20250907.dist-info}/RECORD +32 -25
- torch_geometric/__init__.py +1 -1
- torch_geometric/data/__init__.py +0 -5
- torch_geometric/data/lightning/datamodule.py +2 -2
- torch_geometric/datasets/molecule_gpt_dataset.py +1 -1
- torch_geometric/datasets/web_qsp_dataset.py +262 -210
- torch_geometric/graphgym/imports.py +2 -2
- torch_geometric/llm/__init__.py +9 -0
- torch_geometric/{data → llm}/large_graph_indexer.py +124 -61
- torch_geometric/llm/models/__init__.py +23 -0
- torch_geometric/{nn → llm}/models/g_retriever.py +68 -49
- torch_geometric/{nn → llm}/models/git_mol.py +1 -1
- torch_geometric/{nn/nlp → llm/models}/llm.py +167 -33
- torch_geometric/llm/models/llm_judge.py +158 -0
- torch_geometric/{nn → llm}/models/molecule_gpt.py +1 -1
- torch_geometric/{nn/nlp → llm/models}/sentence_transformer.py +42 -8
- torch_geometric/llm/models/txt2kg.py +353 -0
- torch_geometric/llm/rag_loader.py +154 -0
- torch_geometric/llm/utils/backend_utils.py +442 -0
- torch_geometric/llm/utils/feature_store.py +169 -0
- torch_geometric/llm/utils/graph_store.py +199 -0
- torch_geometric/llm/utils/vectorrag.py +124 -0
- torch_geometric/loader/__init__.py +0 -4
- torch_geometric/nn/__init__.py +0 -1
- torch_geometric/nn/models/__init__.py +0 -10
- torch_geometric/nn/models/sgformer.py +2 -0
- torch_geometric/loader/rag_loader.py +0 -107
- torch_geometric/nn/nlp/__init__.py +0 -9
- {pyg_nightly-2.7.0.dev20250905.dist-info → pyg_nightly-2.7.0.dev20250907.dist-info}/WHEEL +0 -0
- {pyg_nightly-2.7.0.dev20250905.dist-info → pyg_nightly-2.7.0.dev20250907.dist-info}/licenses/LICENSE +0 -0
- /torch_geometric/{nn → llm}/models/glem.py +0 -0
- /torch_geometric/{nn → llm}/models/protein_mpnn.py +0 -0
- /torch_geometric/{nn/nlp → llm/models}/vision_transformer.py +0 -0
{pyg_nightly-2.7.0.dev20250905.dist-info → pyg_nightly-2.7.0.dev20250907.dist-info}/METADATA
RENAMED
@@ -1,6 +1,6 @@
|
|
1
1
|
Metadata-Version: 2.4
|
2
2
|
Name: pyg-nightly
|
3
|
-
Version: 2.7.0.
|
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=
|
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=
|
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=
|
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=
|
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=
|
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=
|
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/
|
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=
|
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=
|
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=
|
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.
|
650
|
-
pyg_nightly-2.7.0.
|
651
|
-
pyg_nightly-2.7.0.
|
652
|
-
pyg_nightly-2.7.0.
|
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,,
|
torch_geometric/__init__.py
CHANGED
@@ -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.
|
34
|
+
__version__ = '2.7.0.dev20250907'
|
35
35
|
|
36
36
|
__all__ = [
|
37
37
|
'Index',
|
torch_geometric/data/__init__.py
CHANGED
@@ -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
|
14
|
+
from lightning.pytorch import LightningDataModule as _LightningDataModule
|
15
15
|
_pl_is_available = True
|
16
16
|
except ImportError:
|
17
17
|
try:
|
18
|
-
from
|
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.
|
17
|
+
from torch_geometric.llm.models import LLM
|
18
18
|
from torch_geometric.utils import one_hot
|
19
19
|
|
20
20
|
|