fix typo in nanogpt python code

feenkcom · Aug 27, 2024 · fde5686 · fde5686
1 parent 948c61f
commit fde5686
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/nanogpt/__init__.py b/python/nanogpt/__init__.py
@@ -171,8 +171,8 @@ def generate(self, idx, max_new_tokens):
 
 @dataclass
 class TrainingContext:
-	model = nn.Module
-	context = torch.Tensor
+	model: nn.Module
+	context: torch.Tensor
 
 	@gtView
 	def gt_view_children(self, builder):