refactor: Remove unused imports and variables

ValerianRey · ValerianRey · commit fa725c3a2ba5 · 2025-12-09T23:27:57.000+01:00
diff --git a/src/recursion/models/common.py b/src/recursion/models/common.py
@@ -1,7 +1,6 @@
 import math
 
 import torch
-from torch import nn
 
 
 def trunc_normal_init_(
diff --git a/src/recursion/models/losses.py b/src/recursion/models/losses.py
@@ -1,4 +1,3 @@
-import math
 from typing import Any, Dict, Optional, Sequence, Tuple
 
 import torch
diff --git a/src/recursion/models/recursive_reasoning/trm.py b/src/recursion/models/recursive_reasoning/trm.py
@@ -284,7 +284,6 @@ def forward(
         input_embeddings = self._input_embeddings(batch["inputs"], batch["puzzle_identifiers"])
 
         # Forward iterations
-        it = 0
         z_H, z_L = carry.z_H, carry.z_L
         # H_cycles-1 without grad
         with torch.no_grad():
diff --git a/src/recursion/models/recursive_reasoning/trm_hier6.py b/src/recursion/models/recursive_reasoning/trm_hier6.py
@@ -354,7 +354,6 @@ def forward(
         input_embeddings = self._input_embeddings(batch["inputs"], batch["puzzle_identifiers"])
 
         # Forward iterations
-        it = 0
         z_H, z_L = carry.z_H, [
             carry.z_L1,
             carry.z_L2,
diff --git a/src/recursion/models/recursive_reasoning/trm_singlez.py b/src/recursion/models/recursive_reasoning/trm_singlez.py
@@ -267,7 +267,6 @@ def forward(
         input_embeddings = self._input_embeddings(batch["inputs"], batch["puzzle_identifiers"])
 
         # Forward iterations
-        it = 0
         z_L = carry.z_L
         # H_cycles-1 without grad
         with torch.no_grad():
diff --git a/src/recursion/puzzle_dataset.py b/src/recursion/puzzle_dataset.py
@@ -5,8 +5,6 @@
 import numpy as np
 import pydantic
 import torch
-from argdantic import ArgParser
-from pydantic import BaseModel
 from torch.utils.data import IterableDataset, get_worker_info
 
 from recursion.dataset.common import PuzzleDatasetMetadata

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`		`-import math`
`2`	`1`	`from typing import Any, Dict, Optional, Sequence, Tuple`
`3`	`2`
`4`	`3`	`import torch`