def __init__()

in realbook/callbacks/utilization.py [0:0]

• 11 lines of code
• 4 McCabe index (conditional complexity)

    def __init__(self, *args: Any, **kwargs: Any):
        """
        Inititialize a GPU utilization callback. This callback will become a no-op
        if a GPU is not present of if nvidia-smi is not installed.
        """
        self.enable = True
        if not nvsmi.is_nvidia_smi_on_path():
            warnings.warn(
                "nvidia-smi is not on path. Please disable this callback. GPU usage data will be unavailable."
            )
            self.enable = False

        if self.enable and len(list(nvsmi.get_gpus())) == 0:
            warnings.warn("No GPUs detected. Please disable this callback. GPU usage data will be unavailable.")
            self.enable = False

        super().__init__(*args, **kwargs)