From a147e6245ca7e3808fcc8b28874431539a554ff1 Mon Sep 17 00:00:00 2001
From: shusank8 <shusank8basyal@gmail.com>
Date: Thu, 23 Oct 2025 11:32:12 -0400
Subject: [PATCH] assert added for Shape mismatch between input and given nin
 value

---
 micrograd/nn.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/micrograd/nn.py b/micrograd/nn.py
index 30d5d777..d612cd51 100644
--- a/micrograd/nn.py
+++ b/micrograd/nn.py
@@ -16,8 +16,10 @@ def __init__(self, nin, nonlin=True):
         self.w = [Value(random.uniform(-1,1)) for _ in range(nin)]
         self.b = Value(0)
         self.nonlin = nonlin
+        self.nin = nin
 
     def __call__(self, x):
+        assert len(x) == self.nin, "Shape mismatch between input and given nin value"
         act = sum((wi*xi for wi,xi in zip(self.w, x)), self.b)
         return act.relu() if self.nonlin else act