def get_optimizer_class()

in optimizers/optimizer.py [0:0]

• 7 lines of code
• 4 McCabe index (conditional complexity)

def get_optimizer_class(optimizer_config: OptimizerConfig):
  if optimizer_config.adam is not None:
    return torch.optim.Adam
  elif optimizer_config.sgd is not None:
    return torch.optim.SGD
  elif optimizer_config.adagrad is not None:
    return torch.optim.Adagrad