fix: Fix local imports

ValerianRey · ValerianRey · commit e2ac1aa96d4a · 2025-12-10T15:32:26.000+01:00
diff --git a/src/recursion/dataset/build_maze_dataset.py b/src/recursion/dataset/build_maze_dataset.py
@@ -6,7 +6,7 @@
 
 import numpy as np
 from argdantic import ArgParser
-from common import PuzzleDatasetMetadata, dihedral_transform
+from recursion.dataset.common import PuzzleDatasetMetadata, dihedral_transform
 from huggingface_hub import hf_hub_download
 from pydantic import BaseModel
 from tqdm import tqdm
diff --git a/src/recursion/dataset/build_sudoku_dataset.py b/src/recursion/dataset/build_sudoku_dataset.py
@@ -5,7 +5,7 @@
 
 import numpy as np
 from argdantic import ArgParser
-from common import PuzzleDatasetMetadata
+from recursion.dataset.common import PuzzleDatasetMetadata
 from huggingface_hub import hf_hub_download
 from pydantic import BaseModel
 from tqdm import tqdm
diff --git a/src/recursion/pretrain.py b/src/recursion/pretrain.py
@@ -15,7 +15,7 @@
 import yaml
 from adam_atan2 import AdamATan2
 from omegaconf import DictConfig
-from puzzle_dataset import PuzzleDataset, PuzzleDatasetConfig, PuzzleDatasetMetadata
+from recursion.puzzle_dataset import PuzzleDataset, PuzzleDatasetConfig, PuzzleDatasetMetadata
 from torch import nn
 from torch.utils.data import DataLoader
 
