[BUG] No module named 'wg_torch'

[maktukmak]

I'm trying to run the example script python/pylibwholegraph/examples/node_classfication.py from the branch-25.10 branch, but it throws the following error:

[rank0]:   File "/usr/lib64/python3.13/runpy.py", line 198, in _run_module_as_main
[rank0]:     return _run_code(code, main_globals, None,
[rank0]:                      "__main__", mod_spec)
[rank0]:   File "/usr/lib64/python3.13/runpy.py", line 88, in _run_code
[rank0]:     exec(code, run_globals)
[rank0]:     ~~~~^^^^^^^^^^^^^^^^^^^
[rank0]:   File "/home/maktukmak/.vscode-server/extensions/ms-python.debugpy-2025.10.0/bundled/libs/debugpy/adapter/../../debugpy/launcher/../../debugpy/__main__.py", line 71, in <module>
[rank0]:     cli.main()
[rank0]:     ~~~~~~~~^^
[rank0]:   File "/home/maktukmak/.vscode-server/extensions/ms-python.debugpy-2025.10.0/bundled/libs/debugpy/adapter/../../debugpy/launcher/../../debugpy/../debugpy/server/cli.py", line 501, in main
[rank0]:     run()
[rank0]:     ~~~^^
[rank0]:   File "/home/maktukmak/.vscode-server/extensions/ms-python.debugpy-2025.10.0/bundled/libs/debugpy/adapter/../../debugpy/launcher/../../debugpy/../debugpy/server/cli.py", line 351, in run_file
[rank0]:     runpy.run_path(target, run_name="__main__")
[rank0]:     ~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]:   File "/home/maktukmak/.vscode-server/extensions/ms-python.debugpy-2025.10.0/bundled/libs/debugpy/_vendored/pydevd/_pydevd_bundle/pydevd_runpy.py", line 310, in run_path
[rank0]:     return _run_module_code(code, init_globals, run_name, pkg_name=pkg_name, script_name=fname)
[rank0]:   File "/home/maktukmak/.vscode-server/extensions/ms-python.debugpy-2025.10.0/bundled/libs/debugpy/_vendored/pydevd/_pydevd_bundle/pydevd_runpy.py", line 127, in _run_module_code
[rank0]:     _run_code(code, mod_globals, init_globals, mod_name, mod_spec, pkg_name, script_name)
[rank0]:     ~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]:   File "/home/maktukmak/.vscode-server/extensions/ms-python.debugpy-2025.10.0/bundled/libs/debugpy/_vendored/pydevd/_pydevd_bundle/pydevd_runpy.py", line 118, in _run_code
[rank0]:     exec(code, run_globals)
[rank0]:     ~~~~^^^^^^^^^^^^^^^^^^^
[rank0]:   File ".test/test_wholegraph2.py", line 272, in <module>
[rank0]:     wgth.distributed_launch(args, main_func)
[rank0]:     ~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
[rank0]:   File "/maktukmak/envs/graph_env/lib64/python3.13/site-packages/pylibwholegraph/torch/distributed_launch.py", line 316, in distributed_launch
[rank0]:     distributed_launch_mpi(args, main_func)
[rank0]:     ~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^
[rank0]:   File "/maktukmak/envs/graph_env/lib64/python3.13/site-packages/pylibwholegraph/torch/distributed_launch.py", line 215, in distributed_launch_mpi
[rank0]:     main_func()
[rank0]:     ~~~~~~~~~^^
[rank0]:   File ".test/test_wholegraph2.py", line 259, in main_func
[rank0]:     wgth.set_framework(args.framework)
[rank0]:     ~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^
[rank0]:   File "/maktukmak/envs/graph_env/lib64/python3.13/site-packages/pylibwholegraph/torch/gnn_model.py", line 34, in set_framework
[rank0]:     from wg_torch.gnn.SAGEConv import SAGEConv
[rank0]: ModuleNotFoundError: No module named 'wg_torch'

After inspecting the codebase, I couldn't find any module named wg_torch defined or installed.

[alexbarghi-nv]

I believe the WG version of SAGEConv is not supported in the open-source release of WholeGraph. Try setting the framework to 'pyg' and I think it should work.