hotfix wandb inpots

Author: drgona

src/neuromancer/trainer.py

@@ -6,20 +6,16 @@
 
 import torch
 from torch.optim.lr_scheduler import ReduceLROnPlateau
-
 import numpy as np
-import wandb
-import lightning.pytorch as pl 
+import lightning.pytorch as pl
 
 from neuromancer.loggers import BasicLogger
 from neuromancer.problem import Problem
 from neuromancer.callbacks import Callback
 from neuromancer.problem import LitProblem
 from neuromancer.dataset import LitDataModule
-
 from lightning.pytorch.callbacks import ModelCheckpoint
 from lightning.pytorch.callbacks.early_stopping import EarlyStopping
-from lightning.pytorch.loggers import WandbLogger
 
 
 

src/neuromancer/utils.py

@@ -1,7 +1,6 @@
 
 import torch
 import functools
-import lightning.pytorch as pl
 from collections import OrderedDict
 
 def handle_device_placement(func):