From 26718967fd1f84270aa7dfd2017097c10f2ebeb4 Mon Sep 17 00:00:00 2001
From: Gaurav S Deshmukh <deshmukg@gilbreth-fe00.rcac.purdue.edu>
Date: Fri, 22 Sep 2023 21:53:28 -0400
Subject: [PATCH] Added TODO

---
 src/models.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/models.py b/src/models.py
index 015d29e..000d01b 100644
--- a/src/models.py
+++ b/src/models.py
@@ -98,6 +98,7 @@ def init_conv_layers(self):
         for i in range(self.n_partitions):
             part_conv_layers = []
             for j in range(self.n_conv[i]):
+                # TODO Add possibility of changing convolutional layers
                 conv_layer = [
                     gnn.CGConv(
                         channels=self.conv_size[i],