Merge pull request #264 from pnnl/master

update develop

Author: drgona

examples/figs/fig_01.png

@@ -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
 
 
 

examples/figs/manifold.png

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