Cleaner way to get options.

cvg · Jan 24, 2024 · 84b9a12 · 84b9a12
1 parent 5863345
commit 84b9a12
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/gluefactory/train.py b/gluefactory/train.py
@@ -153,11 +153,11 @@ def get_lr_scheduler(optimizer, conf):
                 scheduler = get_lr_scheduler(optimizer, scheduler_conf)
                 schedulers.append(scheduler)
 
-            # remove conf.options.schedulers
-            del conf.options.schedulers
+            options = {k: v for k, v in conf.options.items() if k != "schedulers"}
             return getattr(torch.optim.lr_scheduler, conf.type)(
-                optimizer, schedulers, **conf.options
+                optimizer, schedulers, **options
             )
+
         return getattr(torch.optim.lr_scheduler, conf.type)(optimizer, **conf.options)
 
     # backward compatibility