flint fix

rctn · Nov 23, 2024 · 2c58d49 · 2c58d49
1 parent 0211804
commit 2c58d49
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/sparsecoding/transforms/images.py b/sparsecoding/transforms/images.py
@@ -1,5 +1,6 @@
 import torch
 import torch.fft as fft
+import warnings
 from typing import Dict, Optional
 from functools import lru_cache
 from .whiten import whiten, compute_whitening_stats
@@ -281,9 +282,6 @@ def __call__(self, images: torch.Tensor) -> torch.Tensor:
 
     def __repr__(self):
         return "custom whitening augmentation"
-
-import warnings
-import torch
 
 
 def sample_random_patches(