Skip to content

Commit f3c857f

Browse files
committed
Add future todos on kan_block
1 parent a29f954 commit f3c857f

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

pina/_src/model/block/kan_block.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,11 @@
66
from pina._src.model.spline import Spline
77
from pina._src.model.vectorized_spline import VectorizedSpline
88

9-
9+
# TODO
10+
# - Improve documentation and comments throughout the code for better clarity.
11+
# - Remove any unused parameters or code related to the base function if it's
12+
# not being utilized in the current implementation.
13+
# - Clean unused code
1014
class KANBlock(torch.nn.Module):
1115
"""define a KAN layer using splines"""
1216

0 commit comments

Comments
 (0)