From ca4ccddcd6eb60c8a3b04246976269a0eed8c033 Mon Sep 17 00:00:00 2001
From: George Hotz <geohot@gmail.com>
Date: Wed, 12 Jun 2024 09:16:24 +0200
Subject: [PATCH] docsfix: nn.Tensor -> Tensor

---
 tinygrad/nn/state.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tinygrad/nn/state.py b/tinygrad/nn/state.py
index f3c1e0635e..3c098dbff2 100644
--- a/tinygrad/nn/state.py
+++ b/tinygrad/nn/state.py
@@ -40,7 +40,7 @@ def safe_save(tensors:Dict[str, Tensor], fn:str, metadata:Optional[Dict[str, Any
   Saves a state_dict to disk in a .safetensor file with optional metadata.
 
   ```python
-  t = nn.Tensor([1, 2, 3])
+  t = Tensor([1, 2, 3])
   nn.state.safe_save({'t':t}, "test.safetensor")
   ```
   """