diff --git a/docs/api/nn/backbones/index.rst b/docs/api/nn/backbones/index.rst index fd66bee0..01909597 100644 --- a/docs/api/nn/backbones/index.rst +++ b/docs/api/nn/backbones/index.rst @@ -5,9 +5,6 @@ Backbones .. automodule:: topobenchmarkx.nn.backbones.cell.cccn :members: -.. automodule:: topobenchmarkx.nn.backbones.cell.cin - :members: - .. automodule:: topobenchmarkx.nn.backbones.hypergraph.edgnn :members: diff --git a/test/nn/test_auto.py b/test/nn/test_auto.py index 06194d93..ce204d0c 100644 --- a/test/nn/test_auto.py +++ b/test/nn/test_auto.py @@ -1,7 +1,12 @@ +""" +Module for automated testing of neural network modules. + +This module contains functions and utilities to automate the testing +of various neural network modules using the NNModuleAutoTest class. +""" import torch from .._utils.nn_module_auto_test import NNModuleAutoTest from topobenchmarkx.nn.backbones.cell.cccn import CCCN -from topobenchmarkx.nn.backbones.cell.cin import CWN from topobenchmarkx.nn.backbones.hypergraph.edgnn import ( EDGNN, MLP as edgnn_MLP, @@ -11,6 +16,9 @@ def test_auto(): + """ + Function to automate the testing of the modules. + """ num_nodes = 8 d_feat = 12 x = torch.randn(num_nodes, 12) @@ -31,12 +39,6 @@ def test_auto(): "forward": (x, edges_1, edges_2), "assert_shape": (num_nodes, d_feat) }, - #{ - # "module" : CWN, - # "init": (d_feat, d_feat_1, d_feat_2, hid_channels, n_layers), - # "forward": (x, x_1, x_2, edges_1, edges_1, edges_1), - # #"assert_shape": (num_nodes, d_feat) - #}, { "module" : EDGNN, "init": (d_feat, ), diff --git a/topobenchmarkx/nn/backbones/__init__.py b/topobenchmarkx/nn/backbones/__init__.py index 0211a1ce..642c6b63 100644 --- a/topobenchmarkx/nn/backbones/__init__.py +++ b/topobenchmarkx/nn/backbones/__init__.py @@ -2,14 +2,12 @@ from .cell import ( CCCN, - CWN, ) from .hypergraph import EDGNN from .simplicial import SCCNNCustom __all__ = [ "CCCN", - "CWN", "EDGNN", "SCCNNCustom", ] diff --git a/topobenchmarkx/nn/backbones/cell/__init__.py b/topobenchmarkx/nn/backbones/cell/__init__.py index cb628dac..459f48e1 100644 --- a/topobenchmarkx/nn/backbones/cell/__init__.py +++ b/topobenchmarkx/nn/backbones/cell/__init__.py @@ -1,9 +1,7 @@ """Cell backbones.""" from .cccn import CCCN -from .cin import CWN __all__ = [ "CCCN", - "CWN", ]