Skip to content

Commit 468a024

Browse files
authored
Rename batch_size variable to _batch_size
1 parent bfd2bfa commit 468a024

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

‎neural_network/sliding_window_attention.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def forward(self, input_tensor: np.ndarray) -> np.ndarray:
6868
>>> (output.sum() != 0).item() # Check if output is non-zero
6969
True
7070
"""
71-
batch_size, seq_length, _ = input_tensor.shape
71+
_batch_size, seq_length, _ = input_tensor.shape
7272
output = np.zeros_like(input_tensor)
7373

7474
for i in range(seq_length):

0 commit comments

Comments
 (0)