pyg-nightly 2.7.0.dev20250606__py3-none-any.whl → 2.7.0.dev20250608__py3-none-any.whl

{pyg_nightly-2.7.0.dev20250606.dist-info → pyg_nightly-2.7.0.dev20250608.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pyg-nightly
-Version: 2.7.0.dev20250606
+Version: 2.7.0.dev20250608
 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>
@@ -58,7 +58,7 @@ Requires-Dist: torch_geometric[graphgym, modelhub] ; extra == "full"
 Requires-Dist: torchmetrics ; extra == "full"
 Requires-Dist: trimesh ; extra == "full"
 Requires-Dist: protobuf<4.21 ; extra == "graphgym"
-Requires-Dist: pytorch-lightning<2.3.0 ; extra == "graphgym"
+Requires-Dist: pytorch-lightning ; extra == "graphgym"
 Requires-Dist: yacs ; extra == "graphgym"
 Requires-Dist: huggingface_hub ; extra == "modelhub"
 Requires-Dist: pcst_fast ; extra == "rag"
@@ -69,6 +69,7 @@ Requires-Dist: sentencepiece ; extra == "rag"
 Requires-Dist: accelerate ; extra == "rag"
 Requires-Dist: onnx ; extra == "test"
 Requires-Dist: onnxruntime ; extra == "test"
+Requires-Dist: onnxscript ; extra == "test"
 Requires-Dist: pytest ; extra == "test"
 Requires-Dist: pytest-cov ; extra == "test"
 Project-URL: changelog, https://github.com/pyg-team/pytorch_geometric/blob/master/CHANGELOG.md

{pyg_nightly-2.7.0.dev20250606.dist-info → pyg_nightly-2.7.0.dev20250608.dist-info}/RECORD

@@ -1,15 +1,15 @@
-torch_geometric/__init__.py,sha256=BD3Th7gs3RFBES0GFQ42zZ7rQwA2D8n5zYL1L0hvKFs,2255
-torch_geometric/_compile.py,sha256=f-WQeH4VLi5Hn9lrgztFUCSrN_FImjhQa6BxFzcYC38,1338
+torch_geometric/__init__.py,sha256=Tzwv-PYNwUWKL3gK1rsqkdCfWlG2KuPTABN1xaAhZY4,2250
+torch_geometric/_compile.py,sha256=9yqMTBKatZPr40WavJz9FjNi7pQj8YZAZOyZmmRGXgc,1351
 torch_geometric/_onnx.py,sha256=V9ffrIKSqhDw6xUZ12lkuSfNs48cQp2EeJ6Z19GfnVw,349
 torch_geometric/backend.py,sha256=lVaf7aLoVaB3M-UcByUJ1G4T4FOK6LXAg0CF4W3E8jo,1575
 torch_geometric/config_mixin.py,sha256=hOTJu5LLVrEAZ6Pjt4ScLDLKv9aHbfAzF_3ufwKgO4I,4301
 torch_geometric/config_store.py,sha256=zdMzlgBpUmBkPovpYQh5fMNwTZLDq2OneqX47QEx7zk,16818
 torch_geometric/debug.py,sha256=cLyH9OaL2v7POyW-80b19w-ctA7a_5EZsS4aUF1wc2U,1295
-torch_geometric/deprecation.py,sha256=dWRymDIUkUVI2MeEmBG5WF4R6jObZeseSBV9G6FNfjc,858
+torch_geometric/deprecation.py,sha256=gN65uX23c3miRPOpQzxcRS_QDUpD3B-qVKD6y6GX8Yw,872
 torch_geometric/device.py,sha256=tU5-_lBNVbVHl_kUmWPwiG5mQ1pyapwMF4JkmtNN3MM,1224
 torch_geometric/edge_index.py,sha256=RbIwLhtoLXmkQ_DqThBCwi1JH7zNRTsuVj0X-sTYlWE,70094
 torch_geometric/experimental.py,sha256=JbtNNEXjFGI8hZ9raM6-qrZURP6Z5nlDK8QicZUIbz0,4756
-torch_geometric/hash_tensor.py,sha256=YtIGQ29nnR1nXp3NGnm4zR7VpqXteqzyVt36IFtZsHc,24922
+torch_geometric/hash_tensor.py,sha256=ItpMOBtD62zirrNeigZxql0EXofrEcThHX6ZJ50yDYc,24919
 torch_geometric/home.py,sha256=EV54B4Dmiv61GDbkCwtCfWGWJ4eFGwZ8s3KOgGjwYgY,790
 torch_geometric/index.py,sha256=FSP-Tkfw6d7P5U8ua8BO2IoEPJSnJT0L2IKyAfIqJR0,24360
 torch_geometric/inspector.py,sha256=nKi5o4Mn6xsG0Ex1GudTEQt_EqnF9mcMqGtp7Shh9sQ,19336
@@ -19,9 +19,9 @@ torch_geometric/logging.py,sha256=HmHHLiCcM64k-6UYNOSfXPIeSGNAyiGGcn8cD8tlyuQ,85
 torch_geometric/resolver.py,sha256=fn-_6mCpI2xv7eDZnIFcYrHOn0IrwbkWFLDb9laQrWI,1270
 torch_geometric/seed.py,sha256=MJLbVwpb9i8mK3oi32sS__Cq-dRq_afTeoOL_HoA9ko,372
 torch_geometric/template.py,sha256=rqjDWgcSAgTCiV4bkOjWRPaO4PpUdC_RXigzxxBqAu8,1060
-torch_geometric/typing.py,sha256=hASNQY9JZNRGcjvqbGVz32yLpf07etH-cAz4inqQoRI,15622
+torch_geometric/typing.py,sha256=PenHvJZ2ZQ0V6t5hYypcZq8mQ47upgSlVTqHPy6QvLc,15704
 torch_geometric/warnings.py,sha256=SB9dWGovX_KKcxqsOrdTDvSb_j0NoB5vPGnK2vg0jVw,727
-torch_geometric/contrib/__init__.py,sha256=0pWkmXfZtbdr-AKwlii5LTFggTEH-MCrSKpZxrtPlVs,352
+torch_geometric/contrib/__init__.py,sha256=To_lDofgM7sogKEYOICIXei7Njuk-Vkfm-OFhPIdaLo,366
 torch_geometric/contrib/datasets/__init__.py,sha256=lrGnWsEiJf5zsBRmshGZZFN_uYR2ezDjbj9n9nCpvtk,23
 torch_geometric/contrib/explain/__init__.py,sha256=Rs1y07BI6K8J2rmEw6eyrW6QW8y3faaSb3vzWMCoUac,396
 torch_geometric/contrib/explain/pgm_explainer.py,sha256=QglDjv5DXgJMM-37Ozggtz2JLcMYvJgXu7QZMhoHvwQ,16972
@@ -33,28 +33,28 @@ torch_geometric/contrib/transforms/__init__.py,sha256=lrGnWsEiJf5zsBRmshGZZFN_uY
 torch_geometric/data/__init__.py,sha256=D6Iz5A9vEb_2rpf96Zn7uM-lchZ3WpW8X7WdAD1yxKw,4565
 torch_geometric/data/batch.py,sha256=8X8CN4_1rjrh48R3R2--mZUgfsO7Po9JP-H6SbrBiBA,8740
 torch_geometric/data/collate.py,sha256=tOUvttXoEo-bOvJx_qMivJq2JqOsB9iDdjovtiyys4o,12644
-torch_geometric/data/data.py,sha256=mp_jsjsaVwUcY-FghlqNZTHUQEKBdi7xWR_oA2ewrD4,43821
-torch_geometric/data/database.py,sha256=VTct1xyzXsK0GZahBV9-noviCzjRteAsKMG7VgJ52n0,22998
+torch_geometric/data/data.py,sha256=F6DLTd-hJI9hiHemS04Eh-kGZTbg_16-gvRVFsbXfRM,43827
+torch_geometric/data/database.py,sha256=K3KLefYVfsBN9HRItgFZNkbUIllfDt4ueauBFxk3Rxk,23106
 torch_geometric/data/datapipes.py,sha256=9_Cq3j_7LIF4plQFzbLaqyy0LcpKdAic6yiKgMqSX9A,3083
-torch_geometric/data/dataset.py,sha256=rNNAYeJDOQ-a6dqjWg4YKsAz0SvkMZuQK-55sRnmWnk,16769
+torch_geometric/data/dataset.py,sha256=AaJH0N9eZgvxX0ljyTH8cXutKJ0AGFAyE-H4Sw9D51w,16834
 torch_geometric/data/download.py,sha256=kcesTu6jlgmCeePpOxDQOnVhxB_GuZ9iu9ds72KEORc,1889
 torch_geometric/data/extract.py,sha256=X_f0JEo67DF9hOpIlq3QPWXA9RF8uoVFi195UjstzDc,2324
 torch_geometric/data/feature_store.py,sha256=BIMgIWpP1y7OCIQxnkdSWcnm8_BFJXuS_zOqfZZQOjI,20045
 torch_geometric/data/graph_store.py,sha256=EtIgsyY7RdBHRTCn34VypEBOG8cg8WzsNT_kTFKxJR4,13900
-torch_geometric/data/hetero_data.py,sha256=DRmRannOjOffrVnO23EwCavLzAK2sj_pC_-F8d1l14M,48602
-torch_geometric/data/hypergraph_data.py,sha256=-JG3ZnhXluJungr0ELJuwir19DYb7xY_B42k14REKbw,8293
-torch_geometric/data/in_memory_dataset.py,sha256=F35hU9Dw3qiJUL5E1CCAfq-1xrlUMstXBmQVEQdtJ1I,13403
-torch_geometric/data/large_graph_indexer.py,sha256=jSB3St2jT4GUKvmeyr7Hu1ozGk9AQey32Z0XFnkSf4M,25454
+torch_geometric/data/hetero_data.py,sha256=2LV8pSvv-IWkTUk8xg7VeI17YMhikg1RkeQhMwA8tkE,48583
+torch_geometric/data/hypergraph_data.py,sha256=LfriiuJRx9ZrrSGj_fO5NUsh4kvyXJuRdCOqsWo__vc,8307
+torch_geometric/data/in_memory_dataset.py,sha256=ilFxjF4pvBILsS4wOqocwRBc2j6toI2S_KMFF19KB1w,13413
+torch_geometric/data/large_graph_indexer.py,sha256=wlLtc_4wGRuBJmdgf1n-B_JogjEpmT-nnywFOu0D6WE,25440
 torch_geometric/data/makedirs.py,sha256=6uOv4y34i947cm4rv7Aj2_YZBq-EOsyPKnlGA188YSw,463
 torch_geometric/data/on_disk_dataset.py,sha256=77om-e6kzcpBb77kf7um1xY8-yHmQaao_6R7I-3NwHk,6629
 torch_geometric/data/remote_backend_utils.py,sha256=Rzpq1PczXuHhUscrFtIAL6dua6pMehSJlXG7yEsrrrg,4503
 torch_geometric/data/separate.py,sha256=-h4c7QOtGtqbfGXiaMsWARmYYtqiXr6Ie333Sbvu9bs,5587
-torch_geometric/data/storage.py,sha256=iN8uij2A2i6WI69MPLp8b3emZxpbMNfuih1aWXyhZEM,31758
+torch_geometric/data/storage.py,sha256=n5g0GQsuP4tm9nrTIr9BOj1RnOP_TNI_vMCBvEN419E,31772
 torch_geometric/data/summary.py,sha256=o9rmAIkPEAMpsjd4UhoIWtpGsCd0oHpBS99_XkATbIY,5764
 torch_geometric/data/temporal.py,sha256=WOJ6gFrTLikaLhUvotyUF5ql14FkE5Ox3hNkdSp6ZGg,10018
 torch_geometric/data/view.py,sha256=XjkVSc-UWZFCT4DlXLShZtO8duhFQkS9gq88zZXANsk,1089
 torch_geometric/data/lightning/__init__.py,sha256=w3En1tJfy3kSqe1MycpOyZpHFO3fxBCgNCUOznPA3YU,178
-torch_geometric/data/lightning/datamodule.py,sha256=Bn9iaIfE4NWDDWWMqCvBeZ4bIW1Silx_Ol5CPJCliaQ,29242
+torch_geometric/data/lightning/datamodule.py,sha256=IjucsIKRcNv16DIqILQnqa_sz72a4-yivoySmEllv2o,29353
 torch_geometric/datasets/__init__.py,sha256=vIraHnDqD40Num-XwwNivjHQDboK9tmMvlZHjTAuljM,6291
 torch_geometric/datasets/actor.py,sha256=oUxgJIX8bi5hJr1etWNYIFyVQNDDXi1nyVpHGGMEAGQ,4304
 torch_geometric/datasets/airfrans.py,sha256=8cCBmHPttrlKY_iwfyr-K-CUX_JEDjrIOg3r9dQSN7o,5439
@@ -76,7 +76,7 @@ torch_geometric/datasets/coauthor.py,sha256=Nma9aLapDE1S7lCC40WazQZbBJ8nMQV3JJZR
 torch_geometric/datasets/coma.py,sha256=4URaPuXdUJdtZbzWojR-BqxlTyykjtvmXptk3G2Uy9k,4734
 torch_geometric/datasets/cornell.py,sha256=i6wUr2m1U3HCaqMzi-0AZ3Nthdne6_t0ja8qCKYESzE,5311
 torch_geometric/datasets/dblp.py,sha256=Ga59C0CjUQU5ygrDQkW-XmiSYA203sZgR9dTRJlGfIA,5425
-torch_geometric/datasets/dbp15k.py,sha256=w57cWDMz3myrxeoGZg42fx5xKt1mFmbvcggpluvbOx8,5697
+torch_geometric/datasets/dbp15k.py,sha256=oRVnqUPXFIGo_6ufnrELzpr36PCf1hAkGs6k4rTyNfY,5683
 torch_geometric/datasets/deezer_europe.py,sha256=Mi-dOyne7u2oRpO9_8BPv_i3XYb_77XuX1jjD-PSzhg,2511
 torch_geometric/datasets/dgraph.py,sha256=n9jBMWC8KDVa6N-CjxQb2thCk_x_Oh42IM-q7Y0A0KA,4035
 torch_geometric/datasets/dynamic_faust.py,sha256=vPB5Kj5Q0I7mCTpzSR9jaDtEau0mzePi82Ov5cXQrh0,6041
@@ -120,7 +120,7 @@ torch_geometric/datasets/mixhop_synthetic_dataset.py,sha256=4NNvTHUvvV6pcqQCyVDS
 torch_geometric/datasets/mnist_superpixels.py,sha256=o2ArbZ0_OE0u8VCaHmWwvngESlOFr9oM9dSEP_tjAS4,3340
 torch_geometric/datasets/modelnet.py,sha256=-qmLjlQiKVWmtHefAIIE97dQxEcaBfetMJnvgYZuwkg,5347
 torch_geometric/datasets/molecule_gpt_dataset.py,sha256=gVZv14PuZCanE4oxxHlqRNrvzGv6_KN318q5yFA3lS0,18797
-torch_geometric/datasets/molecule_net.py,sha256=VNWLEDulFID8mLsxgN8q1T-O3M2i0n0Si5ISwEZezMU,7379
+torch_geometric/datasets/molecule_net.py,sha256=pMzaJzd-LbBncZ0VoC87HfA8d1F4NwCWTb5YKvLM890,7404
 torch_geometric/datasets/movie_lens.py,sha256=M4Bu0Xus8IkW8GYzjxPxSdPXNbcCCx9cu6cncxBvLx8,4033
 torch_geometric/datasets/movie_lens_100k.py,sha256=eTpBAteM3jqTEtiwLxmhVj4r8JvftvPx8Hvs-3ZIHlU,6057
 torch_geometric/datasets/movie_lens_1m.py,sha256=PRS6G2RYZKgbK3GhUILlwKKDEi3M36fNbHd26Pe8tY8,5396
@@ -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=MbnVCJcryNys1cjdMzXbuk-Rn-BjLbHWFGHx4QYhhUg,14760
+torch_geometric/datasets/tag_dataset.py,sha256=qTnwr2N1tbWYeLGbItfv70UxQ3n1rKesjeVU3kcOCP8,14757
 torch_geometric/datasets/taobao.py,sha256=CUcZpbWsNTasevflO8zqP0YvENy89P7wpKS4MHaDJ6Q,4170
 torch_geometric/datasets/tosca.py,sha256=nUSF8NQT1GlkwWQLshjWmr8xORsvRHzzIqhUyDCvABc,4632
 torch_geometric/datasets/tu_dataset.py,sha256=14OSaXBgVwT1dX2h1wZ3xVIwoo0GQBEfR3yWh6Q0VF0,7847
@@ -159,7 +159,7 @@ torch_geometric/datasets/twitch.py,sha256=qfEerf-Uaojx2ZvegENowdG4E7RoUT_HUO9xtU
 torch_geometric/datasets/upfd.py,sha256=crqO8uQNz1wC1JOn4prSs8iOGv9LuLK3dZf_KUV9tUE,7010
 torch_geometric/datasets/web_qsp_dataset.py,sha256=7ygTK2LGRvJtnlyM3rl-9GV3VpRQWQY_ettKoJRnXz4,11120
 torch_geometric/datasets/webkb.py,sha256=beC1kWeW7cIjYwWyaINQSk-3lmVR85Lus7cKZniHp8Y,4879
-torch_geometric/datasets/wikics.py,sha256=iTzYif1WvbMXnMdhPMfvrkVaAbnM009WiB_f_JWZqhU,3879
+torch_geometric/datasets/wikics.py,sha256=1aELKdZKqqQidkwRSNCJUTtACCbrlZChOuYxakD3fRc,3909
 torch_geometric/datasets/wikidata.py,sha256=9mYShF_HlpTmcdLpiaP_tYJ9eQtUOu5vRPvohN6RXqI,4979
 torch_geometric/datasets/wikipedia_network.py,sha256=a_5o2CAGXak7ceWxwgGcsVBBjpj9zGHpKKDTKiAL7fE,6630
 torch_geometric/datasets/willow_object_class.py,sha256=EXInkHYHRMk8K4noeW0KDYbnxDnppgqI3ZaEWl7ITyc,7028
@@ -190,20 +190,20 @@ torch_geometric/distributed/event_loop.py,sha256=wr3iwMYEWOGkBlvC5huD2k5YxisaGE9
 torch_geometric/distributed/local_feature_store.py,sha256=CLW37RN0ouDufEs2tY9d2nLLvpxubiT6zgW3LIHAU8k,19058
 torch_geometric/distributed/local_graph_store.py,sha256=wNHXSS824Kk2HynbtWFXx-W4pl97UUBv6qFHAVqO3W4,8445
 torch_geometric/distributed/partition.py,sha256=BgjmhDloaooAXM7onGizrcikZs8oRnz5drQZHPDDO_g,14734
-torch_geometric/distributed/rpc.py,sha256=rJqiVR6Vbb2mpyVSC0Y5tPApqP-b1ck1Uq3IQpCsNSw,5737
+torch_geometric/distributed/rpc.py,sha256=j4TZQkk7NB2CIovRrasyvL9l9L4J6_YOq43gpzFMxow,5713
 torch_geometric/distributed/utils.py,sha256=FGrr3qw7hx7EQaIjjqasurloCFJ9q_0jt8jdSIUjBeM,6567
 torch_geometric/explain/__init__.py,sha256=pRxVB33zsxhED1StRWdHboQWh3e06__g9N298Hzi42Y,359
 torch_geometric/explain/config.py,sha256=_0j67NAwPwjrWHPncNywCT-oKyMiryJNxufxVN1BFlM,7834
-torch_geometric/explain/explainer.py,sha256=G7SvraTD25_KZEIP0eQ5QAIuHKodRUF37WOcoAJ921U,10677
+torch_geometric/explain/explainer.py,sha256=eiecPXvwgpRHL6jFp8KumgG0KmF9riJ7dsDEzeAvTCY,10691
 torch_geometric/explain/explanation.py,sha256=5DWNEWhRLptGbnTnMrcugPBHlizUHFlkBx7iwlYo1k4,18883
 torch_geometric/explain/algorithm/__init__.py,sha256=fE29xbd0bPxg-EfrB2BDmmY9QnyO-7TgvYduGHofm5o,496
 torch_geometric/explain/algorithm/attention_explainer.py,sha256=65iGLmOt00ERtBDVxAoydIchykdWZU24aXzSzUGzQEI,11304
 torch_geometric/explain/algorithm/base.py,sha256=lXynh9wMAKWN9tuCkcmsmNPwSUExTtgbNl2Nw39NQD0,6942
 torch_geometric/explain/algorithm/captum.py,sha256=89nHMC-LNp5Mvs_i0F8s1xT_3gmIxcc8izr_PuGmTcY,12858
-torch_geometric/explain/algorithm/captum_explainer.py,sha256=oz-c40hvdzii4_chEQPHzQo_dFjHr9HLuJhDLsqRIVU,7346
+torch_geometric/explain/algorithm/captum_explainer.py,sha256=mgw_m_90UJ18itXXbrE8jnf6I998voWR4l5lJISeoTw,7386
 torch_geometric/explain/algorithm/dummy_explainer.py,sha256=jvcVQmfngmUWgoKa5p7CXzju2HM5D5DfieJhZW3gbLc,2872
 torch_geometric/explain/algorithm/gnn_explainer.py,sha256=CmZZsJEK3PA14NND5CLHmyv37O8VrqHR8Q5faajShVY,24724
-torch_geometric/explain/algorithm/graphmask_explainer.py,sha256=6DisYN-dz2vhr1AlTPBIq6xSmpDdaNu306JKeLObNVI,21377
+torch_geometric/explain/algorithm/graphmask_explainer.py,sha256=CWNQZEwYMPC54JS9BmyT0oJpDToK7yvyqxNLOIrrQZ8,21373
 torch_geometric/explain/algorithm/pg_explainer.py,sha256=LMlNcqSqtEP-IzYA7Xix6FoAogcrLUaEUAxDVyz2eyc,20162
 torch_geometric/explain/algorithm/utils.py,sha256=eh0ARPG41V7piVw5jdMYpV0p7WjTlpehnY-bWqPV_zg,2564
 torch_geometric/explain/metric/__init__.py,sha256=swLeuWVaM3K7UvowsH7q3BzfTq_W1vhcFY8nEP7vFPQ,301
@@ -214,11 +214,11 @@ torch_geometric/graphgym/__init__.py,sha256=XIw3JTK86ca-F5Hf45K_mruHUhbBDC7UGbC0
 torch_geometric/graphgym/benchmark.py,sha256=52Y8fyMQ5Q0hS0kowBlVfnnk-pkLNQLFc8lFs5kiL5Q,510
 torch_geometric/graphgym/checkpoint.py,sha256=v07NKxmXLn0Rqy87t0jHZtyJojRSMWdcwRRdZyWCY18,2408
 torch_geometric/graphgym/cmd_args.py,sha256=l4v_o21YYQ1uaWcKpeMz-D5_CfHKfNDQQbqatomnDgw,738
-torch_geometric/graphgym/config.py,sha256=K_QflMRTvl5MYmHFHuOzXABNxKoySkb6R_jkP10Rm3Y,17197
-torch_geometric/graphgym/imports.py,sha256=IkWtZJN5K8jXul2Gf1_udQsPckteov0noc90_uRM7P0,375
+torch_geometric/graphgym/config.py,sha256=e0TLYKlC_O3VANaYWRfKbjttFb-IhmJayOol5OiPs9g,17210
+torch_geometric/graphgym/imports.py,sha256=hPKDKra8ltQ2gXL_pxajzm7DGHDi6Ty5GkGyYlnaTnE,396
 torch_geometric/graphgym/init.py,sha256=qRlA1BmbtP-ve8XsD4HKpczWRnBMddtI2DdE1pHfJFQ,521
 torch_geometric/graphgym/loader.py,sha256=CEJv8Hw4kwAWr6A7f9GCoU0UIb4K4zWYDx29L7xfiJ8,11605
-torch_geometric/graphgym/logger.py,sha256=QYc41FDhFqZeS6hECxpaFVU3iw7E5RKxsCIFtvOuuVk,11278
+torch_geometric/graphgym/logger.py,sha256=6vqMLOvxPPQcCA2o3W-jaX2PZwZt_rR66NktBwgH48Q,11272
 torch_geometric/graphgym/loss.py,sha256=4pBVhDDohjdrtX9O2SvLWGzyHR1SL62tujcp6Ov-F3Q,1460
 torch_geometric/graphgym/model_builder.py,sha256=R8EEd_8mPWrIcIEYy6J3lb0GGY-KWlCUsAtTBDOd8MQ,3110
 torch_geometric/graphgym/optim.py,sha256=BNlq4XHImeB8QRWiG7kMwksOc3P_SqPGw0AdF2VtZtE,2544
@@ -241,7 +241,7 @@ torch_geometric/graphgym/contrib/train/__init__.py,sha256=4tLhooHLeMNsWUz6c5sndY
 torch_geometric/graphgym/contrib/transform/__init__.py,sha256=4tLhooHLeMNsWUz6c5sndYdE-ken3ucHalY09F2hkHM,221
 torch_geometric/graphgym/models/__init__.py,sha256=0MlCGD-PZNOVYnDR39G8FAT9Jds2EbtXSSjxvlwAGiw,1121
 torch_geometric/graphgym/models/act.py,sha256=PU1sRtzwCKvdavgfdqHNwa1u8wADkBtxdmxKzYGC6FU,855
-torch_geometric/graphgym/models/encoder.py,sha256=Js8r1i34KIjVfltV1ny8kJO-d0zxn92ms7iRifCwcX0,3034
+torch_geometric/graphgym/models/encoder.py,sha256=6fX4x6oAtGRA32A8OBs6MuuFge0PGRNGxTZBL1_s3V0,3006
 torch_geometric/graphgym/models/gnn.py,sha256=ogFkStPm-iOuj_JVNWFWVGvrMG4akWAKtYzNKSYIIuo,6373
 torch_geometric/graphgym/models/head.py,sha256=6OWqehRGvVqtpai30O7Zu305uGbGhwSKpUn-N19eFqk,4603
 torch_geometric/graphgym/models/layer.py,sha256=KIGlWT-BCsbrE68yzcWEiSUI83mllI8X36dSPU58YZU,12499
@@ -250,14 +250,14 @@ torch_geometric/graphgym/models/transform.py,sha256=lbvGYQE0sRx-P5_eTgKbgx2PYWV1
 torch_geometric/graphgym/utils/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 torch_geometric/graphgym/utils/__init__.py,sha256=an9dcDvVAgT3naLq5-Jv5gh2ZkuTlIvaullJIFMbprQ,641
 torch_geometric/graphgym/utils/agg_runs.py,sha256=TGuArA50WcCT8twMVTLyPDRvV3OX8lXMaOKVceGOkPM,9301
-torch_geometric/graphgym/utils/comp_budget.py,sha256=tU5QefRGl3Y9h6fTbnFeqNjJF_cHO_okhwd-bTjxIGY,3047
+torch_geometric/graphgym/utils/comp_budget.py,sha256=hZlCAU56eRa8n6dJ7-ypYtaPjGBwa6bMPHLewUPEKlg,3069
 torch_geometric/graphgym/utils/device.py,sha256=B5lgc3KizEHja3q3Fz22SOENAUXRE8ufCx3aNepBAGk,1342
 torch_geometric/graphgym/utils/epoch.py,sha256=cIlX0Oipx2s4Q8rPJgbLRFQbPqKKNeOynUtGyeyjoFI,690
 torch_geometric/graphgym/utils/io.py,sha256=gqcAXncTCcM9tJpai0LoCokHLoCftnpIgYTPBKARvFY,2049
 torch_geometric/graphgym/utils/plot.py,sha256=UtUL49pLeWXduYgabbDOg0cWLdImVE2yT852fTJqJXw,624
 torch_geometric/graphgym/utils/tools.py,sha256=n-Elfu2_iUncImAWSnWFT-ycMUX9zUJWAKOJ1iHjidE,199
 torch_geometric/io/__init__.py,sha256=3qBVNo84OXjM2vPbOTuKs28i-aAM8b0-i8lD8LpWMdM,528
-torch_geometric/io/fs.py,sha256=dDH_24prUCdnGML20YD0_Ef1-tItvrOdERS63V5jpbw,7618
+torch_geometric/io/fs.py,sha256=O32fpre1zXQ2I0PxPs6zcoZrs955lf6_ujiT2voNWsY,7651
 torch_geometric/io/npz.py,sha256=z4SEfhHvfuADtrZbN0sp7VRcgEZ178Y9xj0XsK7mrCY,1182
 torch_geometric/io/obj.py,sha256=AxGmujEFBr1z7MRXiNeGNM0Trofb4iMsff9IMik0UFs,1088
 torch_geometric/io/off.py,sha256=DGGxxR9w44jmWY81_SlrZpWSdDMz-D2CG8KYOjSzPe0,2757
@@ -276,15 +276,15 @@ torch_geometric/loader/dense_data_loader.py,sha256=GDb_Vu2XyNL5iYzw2zoh1YiurZRr6
 torch_geometric/loader/dynamic_batch_sampler.py,sha256=vjxHDOuctrNjm54hDgsSogZ6q7xRjNVIa6HJ-5VrjHY,4163
 torch_geometric/loader/graph_saint.py,sha256=TLNKmMSE1mekMW_k-wcaWICGx_NONsPe7DicrgMhCYw,8485
 torch_geometric/loader/hgt_loader.py,sha256=1gjYFzn3rU4BlAozRWI0eefUfmi5XC5y6YXqo0VzMw8,6012
-torch_geometric/loader/ibmb_loader.py,sha256=11sg918nIbybr2hoFEO-HA1wYNkL6GFMK9yd8qN6quc,31444
+torch_geometric/loader/ibmb_loader.py,sha256=6OAP_R1tK1Le5MCBYqtZPD0RFsDhfIXDnhgjmlXTAzw,31444
 torch_geometric/loader/imbalanced_sampler.py,sha256=clPERglHRk5SyeFevDrgezYFl7ir975OVFMyJwOV090,3754
 torch_geometric/loader/link_loader.py,sha256=XIYgwo5do_70r-wKQLTKTpET85DMCrEeC3fuKH5xOVQ,16208
 torch_geometric/loader/link_neighbor_loader.py,sha256=HdZifaJI3b0M11Og8m7ztY9YobkE_7fz8aY-dc4QyaM,14384
-torch_geometric/loader/mixin.py,sha256=XX2tIXoFj0dYexUiUhTfVCXAQyL63fpk5bS5FqkXo5U,10946
+torch_geometric/loader/mixin.py,sha256=_2sHZt3rkuWR2iqFMxnz4G9oscR2bt5q7y1hZahlOqY,10976
 torch_geometric/loader/neighbor_loader.py,sha256=vnLn_RhBKTux5h8pi0vzj0d7JPoOpLA3n3vjyIWv9lo,12452
 torch_geometric/loader/neighbor_sampler.py,sha256=mraVFXIIGctYot4Xr2VOAhCKAOQyW2gP9KROf7g6tcc,8497
 torch_geometric/loader/node_loader.py,sha256=g_kV5N0tO6eMSFPc5fdbzfHr4COAeKVJi7FEq52f4zc,11848
-torch_geometric/loader/prefetch.py,sha256=p1mr54TL4nx3Ea0fBy0JulGYJ8Hq4_9rsiNioZsIW-4,3211
+torch_geometric/loader/prefetch.py,sha256=z30TIcu3_6ZubllUOwNLunlq4RyQdFj36vPE5QE_NkM,3232
 torch_geometric/loader/rag_loader.py,sha256=Q2YvFTe4qoThUvsmCQa-4icRvtjBwbTOkWOAQDOhhG4,4642
 torch_geometric/loader/random_node_loader.py,sha256=rCmRXYv70SPxBo-Oh049eFEWEZDV7FmlRPzmjcoirXQ,2196
 torch_geometric/loader/shadow.py,sha256=_hCspYf9SlJYX0lqEjxFec9e9t1iMScNThOoWR1wQGM,4173
@@ -294,9 +294,9 @@ torch_geometric/loader/zip_loader.py,sha256=3lt10fD15Rxm1WhWzypswGzCEwUz4h8OLCD1
 torch_geometric/metrics/__init__.py,sha256=3krvDobW6vV5yHTjq2S2pmOXxNfysNG26muq7z48e94,699
 torch_geometric/metrics/link_pred.py,sha256=dtaI39JB-WqE1B-raiElns6xySRwmkbb9izbcyt6xHI,30886
 torch_geometric/nn/__init__.py,sha256=kQHHHUxFDht2ztD-XFQuv98TvC8MdodaFsIjAvltJBw,874
-torch_geometric/nn/data_parallel.py,sha256=lDAxRi83UNuzAQSj3eu9K2sQheOIU6wqR5elS6oDs90,4764
+torch_geometric/nn/data_parallel.py,sha256=YiybTWoSFyfSzlXAamZ_-y1f7B6tvDEFHOuy_AyJz9Q,4761
 torch_geometric/nn/encoding.py,sha256=QNjwWczYExZ1wRGBmpuqYbn6tB7NC4BU-DEgzjhcZqw,3115
-torch_geometric/nn/fx.py,sha256=oRfnYiih0FM1MhPNcDYIog0oQ0G0soQJuaz7KeNCOjo,16048
+torch_geometric/nn/fx.py,sha256=PDtaHJAgodh4xf8FNl4fVxPGZJDbRaq3Q9z8qb1DNNI,16063
 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
@@ -309,16 +309,16 @@ torch_geometric/nn/sequential.jinja,sha256=yfLxiwVaQdOL2AlGkw2qPdTEKiyQD-m48lCqq
 torch_geometric/nn/sequential.py,sha256=JnfSqN6ZLGb2lZZv8DxafvS4g4G-wVDX9cx-ZpvHqPA,10268
 torch_geometric/nn/summary.py,sha256=Rs59Sr5xMlCROsRNDP-1BuEZOmkvDw1deV7uD-vWXJ8,5822
 torch_geometric/nn/to_fixed_size_transformer.py,sha256=ZjJrWxI0YnoyRjVR7wiOcXZJNnTurt-M6FX6MaVZ0sM,1282
-torch_geometric/nn/to_hetero_module.py,sha256=J2_ZWMnBjsVtwoN5SpykhUHCmhf1qLnBsqngzepprXQ,6519
-torch_geometric/nn/to_hetero_transformer.py,sha256=YS4gFOhnMuDstFTPvR18FDyXNaTxxeDIGaRS9Ubvr6M,18407
-torch_geometric/nn/to_hetero_with_bases_transformer.py,sha256=p6A79eQG1bgOll_mPN-1ptQ6xL3bmC1_hoqQTomnsPQ,22976
+torch_geometric/nn/to_hetero_module.py,sha256=MiKhrDK3EvAsKkGC9Ir8AaB3miFF5X_0vbmEgt2Tuso,6530
+torch_geometric/nn/to_hetero_transformer.py,sha256=NoGC3aglnteyom8zqja__SLm1fZZ37mRI-yNfHagOno,18449
+torch_geometric/nn/to_hetero_with_bases_transformer.py,sha256=ug6A35iVpI-c9QAHyS0oJEC7roYnw3le8jEQfb5a6x0,23018
 torch_geometric/nn/aggr/__init__.py,sha256=_a01GujVyoRSE6-2driodMhC8-jJss4WNIhairYmhHY,1645
 torch_geometric/nn/aggr/attention.py,sha256=ZaZQijQGDx7Mfsk-kDlxJSCDjN1Vp02YyblR5-3SmnY,2952
 torch_geometric/nn/aggr/base.py,sha256=vcWHJlS51mfdYHWmnWEN8b1D57mPM5Y8nqqdVnFm82M,8225
 torch_geometric/nn/aggr/basic.py,sha256=5CRXpm0VVZb22fMPbMMdqQgh97RYiKiUgZGq3hr7Gbw,11011
 torch_geometric/nn/aggr/deep_sets.py,sha256=2LSxJJZaWuxRJew-pubmMYc2ynLYWeTyVK47k6OUhq0,2650
 torch_geometric/nn/aggr/equilibrium.py,sha256=6yY9kelCi1qiSUJPpssR5liGIwS2A9nqi-7K7p8QSfI,6639
-torch_geometric/nn/aggr/fused.py,sha256=AXL8JO47gBIwktfikwGgLmE-OkQ_tKEAuhNP2lOl1nU,12276
+torch_geometric/nn/aggr/fused.py,sha256=JtC5Awm4h2DoLFBmU0fBf96TBTgM4Ba8hsQ0XgE1-Yc,12262
 torch_geometric/nn/aggr/gmt.py,sha256=1JKXHGooA8q_ufGBbfzOhWt693ojuIGJp0lZRqyVNU4,3801
 torch_geometric/nn/aggr/gru.py,sha256=LCLdJWq3xZjJCTxyO3gD8tX6uITQD9U5nCzj8JU4BCA,2193
 torch_geometric/nn/aggr/lcm.py,sha256=TcNqEvHnWpqOc9RFFioBAssQaUhOgMpH1_ovOmgl3ws,4190
@@ -340,7 +340,7 @@ torch_geometric/nn/attention/sgformer.py,sha256=OBC5HQxbY289bPDtwN8UbPH46To2GRTe
 torch_geometric/nn/conv/__init__.py,sha256=8CK-DFG2PEo2ZaFyg-IUlQH8ecQoDDi556uv3ugeQyc,3572
 torch_geometric/nn/conv/agnn_conv.py,sha256=5nEPLx_BBHcDaO6HWzLuHfXc0Yd_reKynAOH0Iq09lU,3077
 torch_geometric/nn/conv/antisymmetric_conv.py,sha256=dhA6sCETy1jlXReYJZBSyToOcL_mZ1wL10fMIb8Ppuw,4387
-torch_geometric/nn/conv/appnp.py,sha256=5hleE5c51Xq0nSP_PyRbr-ukM-3KRROdLrSNhc4AOX0,5983
+torch_geometric/nn/conv/appnp.py,sha256=yM6D1xg-UzUF4KKFHL-vzlnayrh3Jvy2TEwtqIorGkQ,5983
 torch_geometric/nn/conv/arma_conv.py,sha256=5VnImYV9O3Gjxua1YvxB0roc7dEtWY1g4W6HfIZKED0,6585
 torch_geometric/nn/conv/cg_conv.py,sha256=4Jno_6ZhqkzldilMfREXHqgHUVnHcgC1ObxIrFOtrmI,4014
 torch_geometric/nn/conv/cheb_conv.py,sha256=5skDmnURkGk0qYYKwefe6kkj_ZSoivUgk_UQz9siHIU,6423
@@ -350,7 +350,7 @@ torch_geometric/nn/conv/dir_gnn_conv.py,sha256=QWtmSDYKAHSlvRQF4XKhsIw7TFqUNTJS6
 torch_geometric/nn/conv/dna_conv.py,sha256=i4SiBRzZ9a_f1NF_43UCvxd881gfKX1rfxVRWqWWY-g,12242
 torch_geometric/nn/conv/edge_conv.py,sha256=HFcsyOYueRAkkoh6Ifvgj63XcXKkNE3lj4WmKvC9C54,5446
 torch_geometric/nn/conv/edge_updater.jinja,sha256=sCAQMr88Whl4qLqPJ2lGVnBFjllavbHx5N02In81mfk,2216
-torch_geometric/nn/conv/eg_conv.py,sha256=aOKw-ygu8WIhQywiS6tFeNrvPe1W67slhCHa93J3ChQ,10742
+torch_geometric/nn/conv/eg_conv.py,sha256=2MB3Q2qP16i4f0M6zbuzHLLo5z5T1rTzd6vexq-o1mA,10770
 torch_geometric/nn/conv/fa_conv.py,sha256=l7R0QbgG-GnET5wq7W45eaVL6VE0fh_U5VJJ1v-v5PY,9065
 torch_geometric/nn/conv/feast_conv.py,sha256=C5T2-t_3pDW07axRKWL-0WXvtPF6kY8RWBoeqlIKdaU,4430
 torch_geometric/nn/conv/film_conv.py,sha256=Bg9B35RiqK0cFKv6o2Tafm_o5m3NStTygwHmjw4DZjU,6314
@@ -360,22 +360,22 @@ torch_geometric/nn/conv/gated_graph_conv.py,sha256=Rs2uGis5D0XHicixstVUWtramymDU
 torch_geometric/nn/conv/gatv2_conv.py,sha256=_HMu6wNQJzjHEti-7QjlciIwO6L0FXafH94sDfq0jyo,14478
 torch_geometric/nn/conv/gcn2_conv.py,sha256=PUOPOsKL9SuBmFv4FtmiL_XIfkJ_fBPTSzavhfTCmeo,7001
 torch_geometric/nn/conv/gcn_conv.py,sha256=A2tGR2mtHZhQHJ_Bj_MTK1B4kLLFy0Gz5vnh8i1PV_Y,10415
-torch_geometric/nn/conv/gen_conv.py,sha256=RfJ3EWG5_hLvB-ef_Imya_psWy_bCczHEtGdENnYEGk,9722
+torch_geometric/nn/conv/gen_conv.py,sha256=oY5aa9XalFbvoq5oiPXB4Vsh3InIKdi9TwZKSYLDVDQ,9722
 torch_geometric/nn/conv/general_conv.py,sha256=bGyyAtMMS8V8hM7hFizKQC23eVmHWxrwbZZkZgdhotk,7601
 torch_geometric/nn/conv/gin_conv.py,sha256=JYywv9ew72wCuIyJzr4wFHfz00IRJ3a6fn-lvv3w4RU,7411
 torch_geometric/nn/conv/gmm_conv.py,sha256=shTdDOMjTji9PBgduf5NBSYwCh9D4KPu0GjbRiHuv5w,8315
 torch_geometric/nn/conv/gps_conv.py,sha256=uYhq2dbYzOeSdZSzLBiXMYKxj6BPKuyVV5H-nX89SSI,6672
 torch_geometric/nn/conv/graph_conv.py,sha256=J5k6XIOa2JAv-cujKndU5m2_nD1mIoNAIPk0AR6Bgzw,3905
-torch_geometric/nn/conv/gravnet_conv.py,sha256=VgkQqW9VBatx3gcaKKlyYjK75F4UE34YqEDxqJwJTvc,5008
+torch_geometric/nn/conv/gravnet_conv.py,sha256=cuNBx2gKJsmIrsEJ6C2Vz31d0FjwhLSJnmiwIOHXp2k,5038
 torch_geometric/nn/conv/han_conv.py,sha256=mtFkAWKmrmh1pO6IHT1Oa43ueYwQyoAqDH8vTuSnZNg,7198
 torch_geometric/nn/conv/heat_conv.py,sha256=3nU3YDHPQ9-_83WpRsvrJuP1skFGZ-FtjHRaN5aiYbw,6084
-torch_geometric/nn/conv/hetero_conv.py,sha256=lOGxlAs10r986C4FoxQwyxmutCmG-PKiGhINUSP2qI8,6545
+torch_geometric/nn/conv/hetero_conv.py,sha256=eJT_Wmfl0JA1f1r4sHgUScN5CUmP4UZpSPgkOILmcrI,6575
 torch_geometric/nn/conv/hgt_conv.py,sha256=lUhTWUMovMtn9yR_b2-kLNLqHChGOUl2OtXBYISiB8g,9038
 torch_geometric/nn/conv/hypergraph_conv.py,sha256=4BosbbqJyprlI6QjPqIfMxCqnARU_0mUn1zcAQhbw90,8691
 torch_geometric/nn/conv/le_conv.py,sha256=DonmmYZOKk5wIlTZzzIfNKqBY6MO0MRxYhyr0YtNz-Q,3494
 torch_geometric/nn/conv/lg_conv.py,sha256=8jMa79iPsOUbXEfBIc3wmbvAD8T3d1j37LeIFTX3Yag,2369
-torch_geometric/nn/conv/meshcnn_conv.py,sha256=hQpqpOl_pHQA-48tX9E_EDipWcHW2tupI7uiFJfxyHU,22209
-torch_geometric/nn/conv/message_passing.py,sha256=ynTp5MlvHB4SFYnuetK4wWi_1Bj_FhDGAJbf6ZmhEqY,44360
+torch_geometric/nn/conv/meshcnn_conv.py,sha256=Z6p9KwGc_Kj4XQnTWqzbXQzbbpVlMv7ga0DuDB0jLSg,22279
+torch_geometric/nn/conv/message_passing.py,sha256=ZuTvSvodGy1GyAW4mHtuoMUuxclam-7opidYNY5IHm8,44377
 torch_geometric/nn/conv/mf_conv.py,sha256=SkOGMN1tFT9dcqy8xYowsB2ozw6QfkoArgR1BksZZaU,4340
 torch_geometric/nn/conv/mixhop_conv.py,sha256=qVDPWeWcnO7_eHM0ZnpKtr8SISjb4jp0xjgpoDrwjlk,4555
 torch_geometric/nn/conv/nn_conv.py,sha256=X215RSARaJcfI0JOC7K8ybZMq7SoiO_JhJdp9pPRnE8,4759
@@ -391,11 +391,11 @@ torch_geometric/nn/conv/res_gated_graph_conv.py,sha256=JzwfIlRhopZV0d_OxDKfc9cvq
 torch_geometric/nn/conv/rgat_conv.py,sha256=MkDrJIg6fFDbgDi4ibBVyW53MYH3f8rEtX1wmmQEpls,22863
 torch_geometric/nn/conv/rgcn_conv.py,sha256=LU3SQ3M4Rs3Yv1dqpsaIp9GxiZRAZxHy3oXGMKopU7U,15700
 torch_geometric/nn/conv/sage_conv.py,sha256=21pJmDh-sK4u4cs9lXaZk45sD9Gwoci2E1xpHf5a98c,5812
-torch_geometric/nn/conv/sg_conv.py,sha256=3aCbI6bZzNDr7dZ2Q7WmJHPWouNiqSeiODx2EmKG7QY,4543
+torch_geometric/nn/conv/sg_conv.py,sha256=UdKBfc5yL5ll9k_xewVWz5rabf3lOmGgeu92JkYSPcY,4543
 torch_geometric/nn/conv/signed_conv.py,sha256=Q67wZYhmHpZryVYAUv1ycHT1XfVC6U-EUG1uszbm2pU,6190
 torch_geometric/nn/conv/simple_conv.py,sha256=RnlnYIMX1kuKz_ldt8y2M3rPtB79ntpSoVs38Xguilk,3888
-torch_geometric/nn/conv/spline_conv.py,sha256=RzxHKlBAitCRIBbj2Co7MH6HCFwhlzVD2tjNgCdE9Dc,6320
-torch_geometric/nn/conv/ssg_conv.py,sha256=qHjvBwppfazl5qCVsmZz0X1gl9M1fFsVHke9nOArT9w,5131
+torch_geometric/nn/conv/spline_conv.py,sha256=r1-0p1s0BPgEdeoDq1nkUaXy3mf69dQ3Xfec-rKtPBc,6350
+torch_geometric/nn/conv/ssg_conv.py,sha256=i_AkUnVxRjcTP95dc6uGfUskoB6gX9Ne5qcehWK2NDQ,5131
 torch_geometric/nn/conv/supergat_conv.py,sha256=ul0qyLBlMFYFDlwvF_vszqcxiX9hA9eeK9p8AUIAs8M,12420
 torch_geometric/nn/conv/tag_conv.py,sha256=nYcRVQzKqJhe9qmW0QQWHlQielm9AH4gM9AcsiF2ELo,4164
 torch_geometric/nn/conv/transformer_conv.py,sha256=SqtzV9lQwP_Uvmoc428P2ZNdisSWmgu9qq5t8wUV3iQ,10504
@@ -430,45 +430,45 @@ torch_geometric/nn/kge/loader.py,sha256=5Uc1j3OUMQnBYSHDqL7pLCty1siFLzoPkztigYO2
 torch_geometric/nn/kge/rotate.py,sha256=XLuO1AbyTt5cJxr97ZzoyAyIEsHKesgW5TvDmnGJAao,3208
 torch_geometric/nn/kge/transe.py,sha256=jlejq5BLMm-sb1wWcLDp7pZqCdelWBgjDIC8ctbjSdU,3088
 torch_geometric/nn/models/__init__.py,sha256=4mZ5dyiZ9aa1NaBth1qYV-hZdnG_Np1XWvRLB4Qv6RM,2338
-torch_geometric/nn/models/attentive_fp.py,sha256=tkgvw28wg9-JqHIfBllfCwTHrZIUiv85yZJcDqjz3z0,6634
+torch_geometric/nn/models/attentive_fp.py,sha256=1z3iTV2O5W9tqHFAdno8FeBFeXmuG-TDZk4lwwVh3Ac,6634
 torch_geometric/nn/models/attract_repel.py,sha256=h9OyogT0NY0xiT0DkpJHMxH6ZUmo8R-CmwZdKEwq8Ek,5277
 torch_geometric/nn/models/autoencoder.py,sha256=nGje-zty78Y3hxOJ9o0_6QziJjOvBlknk6z0_fDQwQU,10770
 torch_geometric/nn/models/basic_gnn.py,sha256=PGa0RUMyvrNy_5yRI2jX_zwPsmZXwOQWfsWvxOiHsSk,31225
 torch_geometric/nn/models/captum.py,sha256=vPN85_HDMTNcw-rKXAtYY-vT2SbHdf4CFtkseqYsnHg,3972
 torch_geometric/nn/models/correct_and_smooth.py,sha256=wmq-US2r4ocd0a661R8YeDiBeVtILOjdN-4swIth9BQ,6827
-torch_geometric/nn/models/deep_graph_infomax.py,sha256=u6j-5-iHBASDCZ776dyfCI1N8wuvIMjeff8kthiX3Q4,4137
+torch_geometric/nn/models/deep_graph_infomax.py,sha256=yXSZ4mCrq4Dcvl1muzkxEWH4Lo525J4cYuAXpGs55IY,4137
 torch_geometric/nn/models/deepgcn.py,sha256=tIgT03cj8MghYlxEozpoGvGG_CwpJrGDxv1Z0CVIUts,4339
 torch_geometric/nn/models/dimenet.py,sha256=O2rqEx5HWs_lMwRD8eq6WMkbqJaCLL5zgWUJ27qh9Bc,36188
 torch_geometric/nn/models/dimenet_utils.py,sha256=Eyn_EiJqwKvuYj6BtRpSxrzMG3v4Gk98X9MxZ7uvwm4,5069
 torch_geometric/nn/models/g_retriever.py,sha256=tVibbqM_r-1LnA3R3oVyzp0bpuN3qPoYqcU6LZ8dYEk,8260
 torch_geometric/nn/models/git_mol.py,sha256=Wc6Hx6RDDR7sDWRWHfA5eK9e9gFsrTZ9OLmpMfoj3pE,12676
-torch_geometric/nn/models/glem.py,sha256=1Tl9lb5i8-GwtYMiqFtRmgAC7esSr606ekd2RXXBX2Y,16449
+torch_geometric/nn/models/glem.py,sha256=PlLjfMM4lKLs7c7tRC4LVD8tj0jpUXNxcnGbYut7vBE,16624
 torch_geometric/nn/models/gnnff.py,sha256=15dkiLgy0LmH1hnUrpeoHioIp4BPTfjpVATpnGRt9E0,7860
-torch_geometric/nn/models/gpse.py,sha256=3tKzVPkvntAtj13qSPJO1rittr8X1Q_7OqVTOaSpp2s,41982
+torch_geometric/nn/models/gpse.py,sha256=acEAeeicLgzKRL54WhvIFxjA5XViHgXgMEH-NgbMdqI,41971
 torch_geometric/nn/models/graph_mixer.py,sha256=mthMeCOikR8gseEsu4oJ3Cd9C35zHSv1p32ROwnG-6s,9246
-torch_geometric/nn/models/graph_unet.py,sha256=N8TSmJo8AlbZjjcame0xW_jZvMOirL5ahw6qv5Yjpbs,5586
+torch_geometric/nn/models/graph_unet.py,sha256=Zl798oszYNIK9ftvV9L52d_0R9CTftmUlXfJ3rpa90s,5586
 torch_geometric/nn/models/jumping_knowledge.py,sha256=9JR2EoViXKjcDSLb8tjJm-UHfv1mQCJvZAAEsYa0Ocw,5496
 torch_geometric/nn/models/label_prop.py,sha256=6XYBKp7OLmUgcn-73jXXpYZpef9Za1m0gI9QLv2bMEw,3908
 torch_geometric/nn/models/lightgcn.py,sha256=dXKiBvM96jGKLtYp4l90Fko4he0kKdR9fRw-0ZCqSrA,12466
 torch_geometric/nn/models/linkx.py,sha256=BOWGdJ7rJxk4P0XrIy316vyw8PZabbUAK2XKbwYPMQw,5812
 torch_geometric/nn/models/mask_label.py,sha256=B2HcL6ZkaUEo3a8nebZoUqEIfDEfcIGOV56sEmNgOxQ,2580
 torch_geometric/nn/models/meta.py,sha256=lQWovjdQgTGT_rDAm6L186ObINeQCD9tLBz8xenmrF0,6540
-torch_geometric/nn/models/metapath2vec.py,sha256=nxttGe4QVWr4teYEoNz8uHRu-yVsLSZPOeF_tz0bj2o,10788
-torch_geometric/nn/models/mlp.py,sha256=rdwUFxxxqLjXK-iy1L1sXiwSNwAfqTlvHLaqVZ-jwCs,10315
+torch_geometric/nn/models/metapath2vec.py,sha256=jwpEwVUnt9uWcriVfyX1BVCzXOqKpnDVsPQSxJAGoBQ,10788
+torch_geometric/nn/models/mlp.py,sha256=K-WOLQO-DBtaDdMw7zoEbPgItDQg6DCrsd89B6-HvCE,10352
 torch_geometric/nn/models/molecule_gpt.py,sha256=k-XULH6jaurj-R2EE4sIWTkqlNqa3CzWxfQgfFa-G8s,7637
 torch_geometric/nn/models/neural_fingerprint.py,sha256=pTLJgU9Uh2Lnf9bggLj4cKI8YdEFcMF-9MALuubqbuQ,2378
-torch_geometric/nn/models/node2vec.py,sha256=U-VhJlvt5lT-JShFrF5tN84wCPqoVuftLVNyOVXs0OU,7664
+torch_geometric/nn/models/node2vec.py,sha256=81Ku4Rp4IwLEAy06KEgJ2fYtXXVL_uv_Hb8lBr6YXrE,7664
 torch_geometric/nn/models/pmlp.py,sha256=dcAASVSyQMMhItSfEJWPeAFh0R3tNCwAHwdrShwQ8o4,3538
 torch_geometric/nn/models/re_net.py,sha256=pz66q5b5BoGDNVQvpEGS2RGoeKvpjkYAv9r3WAuvITk,8986
 torch_geometric/nn/models/rect.py,sha256=2F3XyyvHTAEuqfJpiNB5M8pSGy738LhPiom5I-SDWqM,2808
-torch_geometric/nn/models/rev_gnn.py,sha256=1b6wU-6YTuLsWn5p8c5LXQm2KugEAVcEYJKZbWTDvgQ,11796
+torch_geometric/nn/models/rev_gnn.py,sha256=Bpme087Zs227lcB0ODOKWsxaly67q96wseaRt6bacjs,11796
 torch_geometric/nn/models/schnet.py,sha256=0aaHrVtxApdvn3RHCGLQJW1MbIb--CSYUrx9O3hDOZM,16656
 torch_geometric/nn/models/sgformer.py,sha256=3NDzkEVRtM1QmeJsXSq7FBhGGchyUvyX1SDPKYg9F70,6875
-torch_geometric/nn/models/signed_gcn.py,sha256=J40CnedFIqtKI1LhW1ITSEFRbA_XiJZL6lASrKwUEAI,9841
+torch_geometric/nn/models/signed_gcn.py,sha256=HEKaXZIWoDnsBRxIytviTpwsjQIFKl44c9glNUpwhlM,9841
 torch_geometric/nn/models/tgn.py,sha256=kEGdfLJybkbMT4UMoAh2nCzfX3_nDjfm1cicuPHEwAM,11878
 torch_geometric/nn/models/visnet.py,sha256=-ABewG-3BzChboHAvoiO3bbi3d_46x8egyvTwxBD5WE,43194
 torch_geometric/nn/nlp/__init__.py,sha256=q6CPUiJHcc9bXw90lyj-ID4F3kfW8uPM-SOxW9uCMHs,213
-torch_geometric/nn/nlp/llm.py,sha256=uVPoIB2_nU0s3N_Z3oVW2Q7XGsAisW1QGaaLe2Y__hw,12227
+torch_geometric/nn/nlp/llm.py,sha256=DAv9jOZKXKQNVU2pNMyS1q8gVUtlin_unc6FjLhOYto,12267
 torch_geometric/nn/nlp/sentence_transformer.py,sha256=q5M7SGtrUzoSiNhKCGFb7JatWiukdhNF6zdq2yiqxwE,4475
 torch_geometric/nn/nlp/vision_transformer.py,sha256=diVBefjIynzYs8WBlcpTeSVnw1PUecHY--B9Yd-W2hA,863
 torch_geometric/nn/norm/__init__.py,sha256=u2qIDrkbeuObGVXSAIftAlvSd6ouGTtxznCfD-59UiA,669
@@ -481,7 +481,7 @@ torch_geometric/nn/norm/layer_norm.py,sha256=m7a7Uoyx0zZhAxJ6Kj5N5DOg4zQnojO1FA1
 torch_geometric/nn/norm/mean_subtraction_norm.py,sha256=KVHOp413mw7obwAN09Le6XdgobtCXpi4UKpjpG1M550,1322
 torch_geometric/nn/norm/msg_norm.py,sha256=zaQtqhs55LU-e6KPC4ylaSdge4KvEoseqOt7pmAzi2s,1662
 torch_geometric/nn/norm/pair_norm.py,sha256=IfHMiVYw_xsy035NakbPGdQVaVC-Ue3Oxwo651Vc47I,2824
-torch_geometric/nn/pool/__init__.py,sha256=2Bi-_xlsGIUUKDeOO7BhaTqCc5n6_ixbu_MO9pglMts,14192
+torch_geometric/nn/pool/__init__.py,sha256=VU9cPdLC-MPgt1kfS0ZwehfSD3g0V30VQuR1Wo0mzZE,14250
 torch_geometric/nn/pool/approx_knn.py,sha256=n7C8Cbar6o5tJcuAbzhM5hqMK26hW8dm5DopuocidO0,3967
 torch_geometric/nn/pool/asap.py,sha256=p8fwpMOeCUyJrdvMmLoTMzr0tI9YCTnefMx8ylIv5xE,6683
 torch_geometric/nn/pool/avg_pool.py,sha256=pwiQh14BCVsT-iULqVAFW-Dxt7DjFOu8CQX_Hu34vZc,3966
@@ -491,7 +491,7 @@ torch_geometric/nn/pool/decimation.py,sha256=AjbU2h_Gl_EQcfkhF977EnrLJ2kait_e4Hy
 torch_geometric/nn/pool/edge_pool.py,sha256=cXgcN5xF8z5NeycYMX9m1zoAk1jtSdyK42YiNNHTeow,8571
 torch_geometric/nn/pool/glob.py,sha256=RJrq1sgAe8oV15WSGtXgB6yXWj2irSJIWAdQLb0byN4,3492
 torch_geometric/nn/pool/graclus.py,sha256=dL9tasXNM-x2NOMRJn8k6z4CeW46nRzoa49IzG58wow,1349
-torch_geometric/nn/pool/knn.py,sha256=fNZV0q2A4lzhZQyePRLHSrtuWjbxQxvv3V7oeNzBLVk,11343
+torch_geometric/nn/pool/knn.py,sha256=dwdTeTiy29cEJilMPoASUvANJIq_rzdCJYFTgbqi3oc,11378
 torch_geometric/nn/pool/max_pool.py,sha256=dOw5f_qP5ALF2oJImC1uQdpim7F8ocLDumP0caRdvFk,4262
 torch_geometric/nn/pool/mem_pool.py,sha256=cl63cGKdEYpOSy-1B6m5N-XZMYe_IzaRrCJyAzwc7AI,5377
 torch_geometric/nn/pool/pan_pool.py,sha256=IoDKils15T568F5vEAny8Zx3vTveVj01fuBEGHbUhi0,4366
@@ -514,24 +514,24 @@ torch_geometric/profile/profile.py,sha256=cHCY4U0XtyqyKC5u380q6TspsOZ5tGHNXaZsKu
 torch_geometric/profile/profiler.py,sha256=rfNciRzWDka_BgO6aPFi3cy8mcT4lSgFWy-WfPgI2SI,16891
 torch_geometric/profile/utils.py,sha256=7h6vzTzW8vv-ZqMOz2DV8HHNgC9ViOrN7IR9d3BPDZ8,5497
 torch_geometric/sampler/__init__.py,sha256=0h_xJ7CQnlTxF5hUpc81WPQ0QaBtouG8eKK1RzPGA-s,512
-torch_geometric/sampler/base.py,sha256=glbuSNcI0A_y-O3x29jqQpSSqSWxUDKJk5vM8uVF8Gs,26980
-torch_geometric/sampler/hgt_sampler.py,sha256=UAm8_wwzEcziKDJ8-TnfZh1705dXRsy_I5PKhZSDTK8,2721
-torch_geometric/sampler/neighbor_sampler.py,sha256=G3pw3IHSAPEVeepiw-vJeWDbQwpEW9jr4-Ms_HjcZCA,34080
+torch_geometric/sampler/base.py,sha256=T7RMx14RSlEKlQUkMvR1EzREaXi14VgR5GIwLfvbXzQ,27055
+torch_geometric/sampler/hgt_sampler.py,sha256=jizRJyEoz4WBOEELuqdytG2hB3UpVQX7yVPM83kvpfE,2991
+torch_geometric/sampler/neighbor_sampler.py,sha256=GBnIwPF5pf85wDo_FaMKtMgBlQhzRx5ysiAomMNhCrQ,34092
 torch_geometric/sampler/utils.py,sha256=RJtasO6Q7Pp3oYEOWrbf2DEYuSfuKZOsF2I7-eJDnoA,5485
 torch_geometric/testing/__init__.py,sha256=m3yp_5UnCAxVgzTFofpiVt0vdbl5GwVAve8WTrAaNxo,1319
 torch_geometric/testing/asserts.py,sha256=DLC9HnBgFWuTIiQs2OalsQcXGhOVG-e6R99IWhkO32c,4606
 torch_geometric/testing/data.py,sha256=O1qo8FyNxt6RGf63Ys3eXBfa5RvYydeZLk74szrez3c,2604
-torch_geometric/testing/decorators.py,sha256=jW2d9qxT7DgWEYFdes2cB-IhnTGg_qjmGufEtHuMX14,9076
-torch_geometric/testing/distributed.py,sha256=ZZCCXqiQC4-m1ExSjDZhS_a1qPXnHEwhJGTmACxNnVI,2227
+torch_geometric/testing/decorators.py,sha256=QhgKfOiA_MXj_X0abWgiF_C5tRygA9HrMddYjEF915M,9097
+torch_geometric/testing/distributed.py,sha256=zSTt5S4euiVreqcbVd_rPshU6A_S9QFCx99seSJdRUM,2227
 torch_geometric/testing/feature_store.py,sha256=J6JBIt2XK-t8yG8B4JzXp-aJcVl5jaCS1m2H7d6OUxs,2158
 torch_geometric/testing/graph_store.py,sha256=00B7QToCIspYmgN7svQKp1iU-qAzEtrt3VQRFxkHfuk,1044
 torch_geometric/transforms/__init__.py,sha256=P0R2CFg9pXxjTX4NnYfNPrifRPAw5lVXEOxO80q-1Ek,4296
-torch_geometric/transforms/add_gpse.py,sha256=ex7cSLmcSIn-dC5gbic-CLwKegV5Is7Y8mUn-mSOWXg,1555
-torch_geometric/transforms/add_metapaths.py,sha256=GabaPRvUnpFrZJsxLMUBY2Egzx94GTgsMxegL_qTtbk,14239
+torch_geometric/transforms/add_gpse.py,sha256=00NTyfP058hdnKKOuakWFdEO0lBkkbzWLqv9ugftJDU,1657
+torch_geometric/transforms/add_metapaths.py,sha256=RAIDpBFJ54GbfPLDmVcs0-IzNfURYbIiAvWta4Gw6JQ,14253
 torch_geometric/transforms/add_positional_encoding.py,sha256=J5dmdrpZ7Qc2n2rcvsAz084uhMGPTtD1xGl973vdiHY,6033
 torch_geometric/transforms/add_remaining_self_loops.py,sha256=ItU5FAcE-mkbp_wqTLkRhv0RShR5JVr8vr9d5xv3_Ak,2085
 torch_geometric/transforms/add_self_loops.py,sha256=No8-tMqERQdWVHwEOaYr9aeg1A_RLisiidEy-1wzoV8,2024
-torch_geometric/transforms/base_transform.py,sha256=5y4X5JmpKrJsj9XQ8v_CYPcDB83pq7b1g5RLjeBrxWg,1298
+torch_geometric/transforms/base_transform.py,sha256=A2P-Xf8v-RDO9b85kKemDcAsvvJD9Hmh8cC-Xg1V2nI,1334
 torch_geometric/transforms/cartesian.py,sha256=_gdFrPP5q3aPmQW6QvYeI8-nvKNVyQF-WEmDpPConww,2464
 torch_geometric/transforms/center.py,sha256=4avx4_wm7Q11epOaMQ2YaVcdcTFjozPBEWG2h6GyKc4,645
 torch_geometric/transforms/compose.py,sha256=P5AFGd6s9L-lpb8io1jKIm2LjAccp_6Q2XocR5T1z5c,1658
@@ -542,20 +542,20 @@ torch_geometric/transforms/face_to_edge.py,sha256=ohAWtpiCs_qHwrrYmz2eLvAcd_bhfP
 torch_geometric/transforms/feature_propagation.py,sha256=GPiKiGU7OuOpBBJeATlCtAtBUy_DSHUoBnJdDK8T81E,3056
 torch_geometric/transforms/fixed_points.py,sha256=sfcqHZSw542LIYmq1DrTJdyncDRa2Uxf5N50G5lYSfQ,2426
 torch_geometric/transforms/gcn_norm.py,sha256=INi8f8J3i2OXWgX5U4GNKROpcvJNW42qO39EdLPRPS8,1397
-torch_geometric/transforms/gdc.py,sha256=d2OGyNrwwOQM7U_C6vTmH6VXyQ1rh-Ig9jTMQN33s4w,20208
+torch_geometric/transforms/gdc.py,sha256=kivoZFHpTJ4Tf_w8fiolkSMP_tl31E-21NbuYRigVis,20245
 torch_geometric/transforms/generate_mesh_normals.py,sha256=eJsmdLPXID7M3xeyQsAjBVugPr4DeSTQ6isheAcoKQs,1028
 torch_geometric/transforms/grid_sampling.py,sha256=aM2NaogaG3XLiKcqZzsDXBnqlJM8tX02qDVaFwY5XmQ,2564
 torch_geometric/transforms/half_hop.py,sha256=xNnLO1J_b7AodwqkAxaZxDr9vScv92LTWLKdwqUQD8k,4102
 torch_geometric/transforms/knn_graph.py,sha256=WnAQ8j89lCH4YJMO0QN_HEQAJOVtWGEKx_8vsH7Hd_E,2536
 torch_geometric/transforms/laplacian_lambda_max.py,sha256=HQpjvmcpePsOv0ZPBJTAAgQiTRnDpwoLsIy--fdbx8E,2505
-torch_geometric/transforms/largest_connected_components.py,sha256=FT0w51NxeAZayLP_GbuUdEjD0_Dmakh4CnsQSUE-Q9s,2161
+torch_geometric/transforms/largest_connected_components.py,sha256=5fqtUNL1CaxTn1tTyV-GkIWNyMwSGCZ15lrL5KT9miM,2161
 torch_geometric/transforms/line_graph.py,sha256=LjhPOs45N_cFWsFseMxYZthwuAjbrE5AW3E6nStSdjs,3730
 torch_geometric/transforms/linear_transformation.py,sha256=PFGCn-91Lyf3t6Gf9lfdiKBhVPXgWkiK5elXWdtfI84,1997
 torch_geometric/transforms/local_cartesian.py,sha256=oF18aDM4i38E8OSMJOgRrHFpalo3cz00N-k_ak5o5L8,2144
 torch_geometric/transforms/local_degree_profile.py,sha256=4haxPEyKKo4qK869jPW0bAwZNOFdlPzdUEjqY2ApKd0,1480
 torch_geometric/transforms/mask.py,sha256=acFjlAsaT-LyThyEM_S_8ghhqotNPjQrhkUAxXw5a9Q,4838
 torch_geometric/transforms/node_property_split.py,sha256=nwK8PT-xWBc4NDIc20ZAK1-S-2SHPvZ2LqIe6ISf-zc,6075
-torch_geometric/transforms/normalize_features.py,sha256=1TCoruyVNLbrD3xuZv4V98rwGPvp_C1xKTqCFnQS6R0,1028
+torch_geometric/transforms/normalize_features.py,sha256=RmljMqn7t9UC-n2CU9an7BnTlsgKWujpY5e9nz4mh9Q,1064
 torch_geometric/transforms/normalize_rotation.py,sha256=S_r1G5OtBlCjAmP8plM6m2qoM52J64tv7vGQPIvD8kY,1782
 torch_geometric/transforms/normalize_scale.py,sha256=Q_fKOdZzJ7bLlgraDeRuSAIw_BhsXNOrYqINqhvh6Cw,666
 torch_geometric/transforms/one_hot_degree.py,sha256=ITKs4QMaf6dx4e5GYup316kkYhCX0-U1XTZ9EAMec4U,1584
@@ -565,12 +565,12 @@ torch_geometric/transforms/polar.py,sha256=MtWYzXfyK2-_PRV-rwKUBAQAXgXAWEkE6eQcm
 torch_geometric/transforms/radius_graph.py,sha256=gkLm2rPoEluTyXxkW48-Vy1dPUS7QRCInVjod0fbIXw,2043
 torch_geometric/transforms/random_flip.py,sha256=uPR7v2goD0qHIZmXfY8YzIYzWF86weyVx_iXM_W53yk,1033
 torch_geometric/transforms/random_jitter.py,sha256=h2ZSF7zD_pUTRdAXjMD1DIiCbd3nU0uZLOOw6imX29I,1721
-torch_geometric/transforms/random_link_split.py,sha256=OuS507qsYZhdHILPKEicOgqqu0-SuTqcaQvw7wm4FIc,15181
+torch_geometric/transforms/random_link_split.py,sha256=uk_uNeU-J9jUoIphvyj6mE3rBhEiZ_50KfrKq8fhsQY,15195
 torch_geometric/transforms/random_node_split.py,sha256=1B9lEb_AKMpIWCe3l51FO8GKVz8bI6Q5KbsALGNZWMM,5853
 torch_geometric/transforms/random_rotate.py,sha256=5DIHYmsu2FQuZnwsQMQYD1msMnJf_C1FWQHKSdkEsrA,1946
 torch_geometric/transforms/random_scale.py,sha256=Qc0ciXDKQ8CEqOnacePyT0WLtOO74L2ABsiGRWYAVmU,1261
 torch_geometric/transforms/random_shear.py,sha256=TXnxdrvLRNdsi9NIYev9941MvrW2SFdYs1FFfAjV26A,1365
-torch_geometric/transforms/remove_duplicated_edges.py,sha256=EMX6E1R_gXFoXwBqMqt6vA930mK0unqFu56-BQUf9aY,1929
+torch_geometric/transforms/remove_duplicated_edges.py,sha256=FpI6omjr_QWLFj7WCNMbhrJZXu0e0qp-dXH9e9xBNd8,1976
 torch_geometric/transforms/remove_isolated_nodes.py,sha256=Q89b73es1tPsAmTdS7tWTIM7JcPUpL37v3EZTAd25Fc,2449
 torch_geometric/transforms/remove_self_loops.py,sha256=JfoooSnTO2KPXXuC3KWGhLS0tqr4yiXOl3A0sVv2riM,1221
 torch_geometric/transforms/remove_training_classes.py,sha256=GMCZwI_LYo2ZF29DABZXeuM0Sn2i3twx_V3KBUGu2As,932
@@ -599,12 +599,12 @@ torch_geometric/utils/_negative_sampling.py,sha256=jxsmpryeoTT8qQrvIH11MgyhgoWzv
 torch_geometric/utils/_normalize_edge_index.py,sha256=H6DY-Dzi1Psr3igG_nb0U3ZPNZz-BBDntO2iuA8FtzA,1682
 torch_geometric/utils/_normalized_cut.py,sha256=uwVJkl-Q0tpY-w0nvcHajcQYcqFh1oDOf55XELdjJBU,1167
 torch_geometric/utils/_one_hot.py,sha256=vXC7l7zudYRZIwWv6mT-Biuk2zKELyqteJXLynPocPM,1404
-torch_geometric/utils/_scatter.py,sha256=qwM8GpMiXT8uuckTg9nYDtwg2WqAWljkhjLE8omOHhk,11642
+torch_geometric/utils/_scatter.py,sha256=FPt5T41wbg0ilLMtAcJR4SwPqlxDrjDqO7KgA-Ynbwo,11672
 torch_geometric/utils/_segment.py,sha256=CqS7_NMQihX89gEwFVHbyMEZgaEnSlJGpyuWqy3i8HI,1976
 torch_geometric/utils/_select.py,sha256=BZ5P6-1riR4xhCIJZnsNg5HmeAGelRzH42TpADj9xpQ,2439
 torch_geometric/utils/_softmax.py,sha256=6dTVbWX04laemRP-ZFPMS6ymRZtRa8zYF22QCXl_m4w,3242
 torch_geometric/utils/_sort_edge_index.py,sha256=Z5F9xcRp3hKzTiTlc2gqYufs8QLDSI4cuJNRI0zF0G4,4373
-torch_geometric/utils/_spmm.py,sha256=zesnYzhGZoEih99iW_fMBysiWFPLLiuV8YTIQg0fOL4,5740
+torch_geometric/utils/_spmm.py,sha256=2fBcllo-ywstsd8mfoMwSP0tyI_lBU033P1Yd9Bhuz8,5743
 torch_geometric/utils/_subgraph.py,sha256=GcOGNUcVe97tifuQyi5qBZ88A_Wo3-o17l9xCSIsau4,18456
 torch_geometric/utils/_to_dense_adj.py,sha256=hl1sboUBvED5Er66bqLms4VdmxKA-7Y3ozJIR-YIAUc,3606
 torch_geometric/utils/_to_dense_batch.py,sha256=-K5NjjfvjKYKJQ3kXgNIDR7lwMJ_GGISI45b50IGMvY,4582
@@ -613,12 +613,12 @@ torch_geometric/utils/_tree_decomposition.py,sha256=ZtpjPQJgXbQWtSWjo-Fmhrov0DGO
 torch_geometric/utils/_trim_to_layer.py,sha256=cauOEzMJJK4w9BC-Pg1bHVncBYqG9XxQex3rn10BFjc,8339
 torch_geometric/utils/_unbatch.py,sha256=B0vjKI96PtHvSBG8F_lqvsiJE134aVjUurPZsG6UZRI,2378
 torch_geometric/utils/augmentation.py,sha256=1F0YCuaklZ9ZbXxdFV0oOoemWvLd8p60WvFo2chzl7E,8600
-torch_geometric/utils/convert.py,sha256=0KEJoBOzU-w-mMQu9QYaMhUqcrGBxBmeRl0hv8NPvII,21697
+torch_geometric/utils/convert.py,sha256=j0t_87c-U_-15YKFfkOZfloEc5NbjgeLIk851zHG8WA,21665
 torch_geometric/utils/cross_entropy.py,sha256=ZFS5bivtzv3EV9zqgKsekmuQyoZZggPSclhl_tRNHxo,3047
 torch_geometric/utils/dropout.py,sha256=gg0rDnD4FLvBaKSoLAkZwViAQflhLefJm6_Mju5dmQs,11416
-torch_geometric/utils/embedding.py,sha256=b-CQ-aapEgahxSS7fuL4aNQX6GJROboV0xclZ_MmwO0,5179
+torch_geometric/utils/embedding.py,sha256=Ac_MPSrZGpw-e-gU6Yz-seUioC2WZxBSSzXFeclGwMk,5232
 torch_geometric/utils/functions.py,sha256=orQdS_6EpzWSmBHSok3WhxCzLy9neB-cin1aTnlXY-8,703
-torch_geometric/utils/geodesic.py,sha256=-xsqE3FZU7Y9gMbucIlGJ4FM-3nk8o0AQBxIdN-QfEw,4770
+torch_geometric/utils/geodesic.py,sha256=e_XCn7dxqeYJBL-sAc2DfxF3kp_ZUIP0vwqsx1yshmU,4777
 torch_geometric/utils/hetero.py,sha256=ok4uAAOyMiaeEPmvyS4DNoDwdKnLS2gmgs5WVVklxOo,5539
 torch_geometric/utils/influence.py,sha256=7R-NW4myJMJPkbNiwcHTmO_m_B3gPB2IlBbQkB446xc,10348
 torch_geometric/utils/isolated.py,sha256=nUxCfMY3q9IIFjelr4eyAJH4sYG9W3lGdpWidnp3dm4,3588
@@ -635,12 +635,12 @@ torch_geometric/utils/ppr.py,sha256=ebiHbQqRJsQbGUI5xu-IkzQSQsgIaC71vgO0KcXIKAk,
 torch_geometric/utils/random.py,sha256=Rv5HlhG5310rytbT9EZ7xWLGKQfozfz1azvYi5nx2-U,5148
 torch_geometric/utils/repeat.py,sha256=RxCoRoEisaP6NouXPPW5tY1Rn-tIfrmpJPm0qGP6W8M,815
 torch_geometric/utils/smiles.py,sha256=lGQ2BwJ49uBrQfIxxPz8ceTO9Jo-XCjlLxs1ql3xrsA,7130
-torch_geometric/utils/sparse.py,sha256=MJyWkn-r9sdMyR_m-aBUIQUkvxsYLUNP9jYlfntSUpI,25118
+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.dev20250606.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
-pyg_nightly-2.7.0.dev20250606.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-pyg_nightly-2.7.0.dev20250606.dist-info/METADATA,sha256=DZmmGP4X5ZCOarKh3B4kkaqENUf31OB0kSOV_zeizTI,62967
-pyg_nightly-2.7.0.dev20250606.dist-info/RECORD,,
+pyg_nightly-2.7.0.dev20250608.dist-info/licenses/LICENSE,sha256=ic-27cMJc1kWoMEYncz3Ya3Ur2Bi3bNLWib2DT763-o,1067
+pyg_nightly-2.7.0.dev20250608.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+pyg_nightly-2.7.0.dev20250608.dist-info/METADATA,sha256=Sa3pjWIE537hdoBeNbKNVZKsAUxLXwqpQ7lXhtJcap0,63005
+pyg_nightly-2.7.0.dev20250608.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.dev20250606'
+__version__ = '2.7.0.dev20250608'
 
 __all__ = [
     'Index',
@@ -60,9 +60,10 @@ __all__ = [
 if not torch_geometric.typing.WITH_PT113:
     import warnings as std_warnings
 
-    std_warnings.warn("PyG 2.7 removed support for PyTorch < 1.13. Consider "
-                      "Consider upgrading to PyTorch >= 1.13 or downgrading "
-                      "to PyG <= 2.6. ")
+    std_warnings.warn(
+        "PyG 2.7 removed support for PyTorch < 1.13. Consider "
+        "Consider upgrading to PyTorch >= 1.13 or downgrading "
+        "to PyG <= 2.6. ", stacklevel=2)
 
 # Serialization ###############################################################
 

torch_geometric/_compile.py

@@ -36,6 +36,7 @@ def compile(
         :meth:`torch_geometric.compile` is deprecated in favor of
         :meth:`torch.compile`.
     """
-    warnings.warn("'torch_geometric.compile' is deprecated in favor of "
-                  "'torch.compile'")
+    warnings.warn(
+        "'torch_geometric.compile' is deprecated in favor of "
+        "'torch.compile'", stacklevel=2)
     return torch.compile(model, *args, **kwargs)  # type: ignore

torch_geometric/contrib/__init__.py

@@ -7,6 +7,6 @@ import torch_geometric.contrib.explain  # noqa
 
 warnings.warn(
     "'torch_geometric.contrib' contains experimental code and is subject to "
-    "change. Please use with caution.")
+    "change. Please use with caution.", stacklevel=2)
 
 __all__ = []

torch_geometric/data/data.py

@@ -850,14 +850,14 @@ class Data(BaseData, FeatureStore, GraphStore):
         # that maps global node indices to local ones in the final
         # heterogeneous graph:
         node_ids, index_map = {}, torch.empty_like(node_type)
-        for i, key in enumerate(node_type_names):
+        for i in range(len(node_type_names)):
             node_ids[i] = (node_type == i).nonzero(as_tuple=False).view(-1)
             index_map[node_ids[i]] = torch.arange(len(node_ids[i]),
                                                   device=index_map.device)
 
         # We iterate over edge types to find the local edge indices:
         edge_ids = {}
-        for i, key in enumerate(edge_type_names):
+        for i in range(len(edge_type_names)):
             edge_ids[i] = (edge_type == i).nonzero(as_tuple=False).view(-1)
 
         data = HeteroData()
@@ -1193,4 +1193,4 @@ def warn_or_raise(msg: str, raise_on_error: bool = True):
     if raise_on_error:
         raise ValueError(msg)
     else:
-        warnings.warn(msg)
+        warnings.warn(msg, stacklevel=2)

torch_geometric/data/database.py

@@ -111,13 +111,17 @@ class Database(ABC):
             for key, value in schema_dict.items()
         }
 
+    @abstractmethod
     def connect(self) -> None:
         r"""Connects to the database.
         Databases will automatically connect on instantiation.
         """
+        raise NotImplementedError
 
+    @abstractmethod
     def close(self) -> None:
         r"""Closes the connection to the database."""
+        raise NotImplementedError
 
     @abstractmethod
     def insert(self, index: int, data: Any) -> None: