fix code formatting under modules/dml

This commit is contained in:
Seunghoon Lee
2024-02-05 22:43:10 +09:00
parent ef29f1a238
commit ff2c1db1cc
19 changed files with 91 additions and 46 deletions
+1
View File
@@ -1,6 +1,7 @@
from typing import Optional, Union
import torch
rDevice = Union[torch.device, int]
def get_device(device: Optional[rDevice]=None) -> torch.device:
if device is None: