geekyrakshit commited on
Commit
60407c7
1 Parent(s): 8ff8f45

updated mirnet

Browse files
Files changed (1) hide show
  1. enhance_me/mirnet/mirnet.py +1 -1
enhance_me/mirnet/mirnet.py CHANGED
@@ -21,7 +21,7 @@ from ..commons import (
21
 
22
 
23
  class MIRNet:
24
- def __init__(self, experiment_name: str, wandb_api_key=None) -> None:
25
  self.experiment_name = experiment_name
26
  if wandb_api_key is not None:
27
  init_wandb("mirnet", experiment_name, wandb_api_key)
 
21
 
22
 
23
  class MIRNet:
24
+ def __init__(self, experiment_name=None, wandb_api_key=None) -> None:
25
  self.experiment_name = experiment_name
26
  if wandb_api_key is not None:
27
  init_wandb("mirnet", experiment_name, wandb_api_key)