From db7edf86af56aa76eaaa2b30f3b6568e401c7ecd Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Tue, 12 Jan 2021 11:38:14 +0800 Subject: Remove TypeDict for python 3.7 --- utils/misc.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'utils/misc.py') 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: -- cgit v1.2.3