def _create_runnable()

in python-batch/batch.py [0:0]

• 17 lines of code
• 3 McCabe index (conditional complexity)

  def _create_runnable(self) -> batch_v1.Runnable:

    self.runnable = batch_v1.Runnable()
    self.runnable.environment = batch_v1.Environment()
    self.runnable.environment.variables = self.env_vars

    if "container" in self.config:
      self.runnable.container = batch_v1.Runnable.Container()
      self.runnable.container.image_uri = self.config["container"]["image_uri"] 
      self.runnable.container.entrypoint =  self.config["container"]["entry_point"] 
      self.runnable.container.commands =  self.config["container"]["commands"] 
      self.runnable.container.options = "--privileged"
      if "install_gpu_drivers" in self.config:
        self.runnable.container.volumes.append("/var/lib/nvidia/lib64:/usr/local/nvidia/lib64")
        self.runnable.container.volumes.append("/var/lib/nvidia/bin:/usr/local/nvidia/bin")

    else:
      self.runnable.script = batch_v1.Runnable.Script()
      self.runnable.script.text = self.config["script_text"]
    return(self.runnable)