Browse Source

[MNT] solve E266 error of flake8 in package

tags/v0.3.2
Gene 1 year ago
parent
commit
7b1960659b
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      learnware/__init__.py

+ 3
- 3
learnware/__init__.py View File

@@ -36,12 +36,12 @@ def init(verbose=True, **kwargs):
with open(config_file, "r") as fin_config:
C.update(**dict(json.load(fin_config)))

## random seed
# random seed
deterministic = kwargs.get("deterministic", True)
if deterministic:
setup_seed(C.random_seed)

## make dirs
# make dirs
mkdir = kwargs.get("mkdir", True)
if mkdir:
os.makedirs(C.root_path, exist_ok=True)
@@ -49,7 +49,7 @@ def init(verbose=True, **kwargs):
os.makedirs(C.stdout_path, exist_ok=True)
os.makedirs(C.cache_path, exist_ok=True)

## ignore tensorflow warning
# ignore tensorflow warning
tf_loglevel = kwargs.get("tf_loglevel", "2")
os.environ["TF_CPP_MIN_LOG_LEVEL"] = tf_loglevel



Loading…
Cancel
Save