(gpu_to_use: int | None, device: str | None)
| 1986 | |
| 1987 | |
| 1988 | def _gpu_to_use_to_device(gpu_to_use: int | None, device: str | None) -> str | None: |
| 1989 | if device is None and gpu_to_use is not None: |
| 1990 | if isinstance(gpu_to_use, int): |
| 1991 | device = f"cuda:{gpu_to_use}" |
| 1992 | else: |
| 1993 | device = gpu_to_use |
| 1994 | |
| 1995 | return device |
| 1996 | |
| 1997 | |
| 1998 | def _load_config(config: str) -> dict: |
no outgoing calls
no test coverage detected