fix main (#186)

* undo main

* update

Author: aniketmaurya

gradsflow/autotasks/engine/backend.py

@@ -13,7 +13,6 @@
 #  limitations under the License.
 
 import logging
-import math
 import typing
 from enum import Enum
 from typing import Callable, Dict, Optional
@@ -28,7 +27,6 @@
 if typing.TYPE_CHECKING:
     import pytorch_lightning as pl
 
-
 if is_installed("pytorch_lightning"):
     from flash import Task
     from flash import Trainer as FlashTrainer
@@ -114,7 +112,9 @@ def _lightning_objective(
         logger.debug(trainer.callback_metrics)
         return trainer.callback_metrics[self.optimization_metric].item()
 
-    def optimization_objective(self, config: dict, trainer_config: dict, finetune: bool, gpu: Optional[float] = 0.0):
+    def optimization_objective(
+        self, config: dict, trainer_config: dict, finetune: bool = False, gpu: Optional[float] = 0.0
+    ):
         """
         Defines lightning_objective function which is used by tuner to minimize/maximize the metric.