From 1b8aacba29f6941e6612423a60595a91fb9d6192 Mon Sep 17 00:00:00 2001
From: levtelyatnikov <tlscabinet@gmail.com>
Date: Tue, 4 Jun 2024 20:51:03 +0200
Subject: [PATCH 1/3] updated docs api

---
 docs/api/nn/backbones/index.rst | 3 ---
 1 file changed, 3 deletions(-)

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:
 

From 6190360cfa13f4c6a83a276bd25acd031b03b4ab Mon Sep 17 00:00:00 2001
From: levtelyatnikov <tlscabinet@gmail.com>
Date: Tue, 4 Jun 2024 21:00:17 +0200
Subject: [PATCH 2/3] updated backbones/init.py

---
 topobenchmarkx/nn/backbones/__init__.py | 2 --
 1 file changed, 2 deletions(-)

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",
 ]

From 65132c4993303e6eb38ed51ef80e8164f12e23b3 Mon Sep 17 00:00:00 2001
From: levtelyatnikov <tlscabinet@gmail.com>
Date: Tue, 4 Jun 2024 21:11:21 +0200
Subject: [PATCH 3/3] fixing doc string of test_auto

---
 test/nn/test_auto.py                         | 16 +++++++++-------
 topobenchmarkx/nn/backbones/cell/__init__.py |  2 --
 2 files changed, 9 insertions(+), 9 deletions(-)

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/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",
 ]