diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-12 11:38:14 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-12 11:38:31 +0800 |
| commit | db7edf86af56aa76eaaa2b30f3b6568e401c7ecd (patch) | |
| tree | b6e28b09831b70bb81a116b49b9c47c900376e11 /utils/misc.py | |
| parent | 966d4431c037b0c4641aa2a5fc22f05be064b331 (diff) | |
Remove TypeDict for python 3.7
Diffstat (limited to 'utils/misc.py')
| -rw-r--r-- | utils/misc.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/utils/misc.py b/utils/misc.py index b850830..c047618 100644 --- a/utils/misc.py +++ b/utils/misc.py @@ -1,9 +1,8 @@ import os +from typing import Dict -from utils.configuration import SystemConfiguration - -def set_visible_cuda(config: SystemConfiguration): +def set_visible_cuda(config: Dict): """Set environment variable CUDA device(s)""" CUDA_VISIBLE_DEVICES = config.get('CUDA_VISIBLE_DEVICES', None) if CUDA_VISIBLE_DEVICES: |