class_docstr: Extend :class:`DartsLightningModule` to support gumbel-softmax with temperature annealing.
  
  The default implementation of :class:`~nni.nas.strategy.GumbelDARTS`.
  
  See Also
  --------
  nni.nas.strategy.GumbelDARTS
  DartsLightningModule
class_docstr_tok: ["Extend",":","class",":","`","DartsLightningModule","`","to","support","gumbel-softmax","with","temperature","annealing",".","The","default","implementation","of",":","class",":","`","~nni.nas.strategy.GumbelDARTS","`",".","See","Also","--","--","--","--","nni.nas.strategy.GumbelDARTS","DartsLightningModule"]
code: def __init__(
      self,
      training_module: pl.LightningModule,
      temperature_scheduler: LinearTemperatureScheduler,
      *args,
      **kwargs,
  ):
      super().__init__(training_module, *args, **kwargs)
      self.temperature_scheduler = temperature_scheduler
  
imports: from __future__ import annotations
  import logging
  import pytorch_lightning
  import torch
  import torch.nn
  import torch.optim
  from .base_lightning import BaseOneShotLightningModule
  from .supermodule.differentiable import GumbelSoftmax
  from .profiler import ExpectationProfilerPenalty
label_desc: Use image node_id 1 to create a new GumbelDartsLightningModule object from inherited base classes: DartsLightningModule with example: obj = GumbelDartsLightningModule(training_module, temperature_scheduler)