fix: import order

berenslab · Oct 21, 2024 · 5c13643 · 5c13643
1 parent 9744025
commit 5c13643
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/runner/analyze.py b/runner/analyze.py
@@ -5,10 +5,10 @@
 
 import matplotlib.pyplot as plt
 import torch
+import wandb
 from matplotlib.figure import Figure
 from omegaconf import DictConfig
 
-import wandb
 from retinal_rl.analysis.plot import (
     layer_receptive_field_plots,
     plot_brain_and_optimizers,

diff --git a/runner/initialize.py b/runner/initialize.py
@@ -7,11 +7,11 @@
 
 import omegaconf
 import torch
+import wandb
 from hydra.core.hydra_config import HydraConfig
 from omegaconf import DictConfig
 from torch.optim.optimizer import Optimizer
 
-import wandb
 from retinal_rl.models.brain import Brain
 from runner.util import save_checkpoint
 

diff --git a/runner/sweep.py b/runner/sweep.py
@@ -2,9 +2,8 @@
 
 from typing import Any, Dict, cast
 
-from omegaconf import DictConfig, OmegaConf
-
 import wandb
+from omegaconf import DictConfig, OmegaConf
 
 
 def launch_sweep(cfg: DictConfig):