From 3a006ba50ecfcf1349af1f4ce54afca614918aa0 Mon Sep 17 00:00:00 2001 From: pzhokhov Date: Tue, 21 Aug 2018 17:41:25 -0700 Subject: [PATCH] flake8 fixes (#35) * flake8 fixes * added baselines/setup.cfg * style checks using setup.cfg in baselines --- .travis.yml | 2 +- baselines/run.py | 3 +-- setup.cfg | 15 +++++++++++++++ setup.py | 1 - 4 files changed, 17 insertions(+), 4 deletions(-) create mode 100644 setup.cfg diff --git a/.travis.yml b/.travis.yml index d4e9b26..f7ade0a 100644 --- a/.travis.yml +++ b/.travis.yml @@ -10,5 +10,5 @@ install: - docker build . -t baselines-test script: - - flake8 --select=F,E999 baselines/common baselines/trpo_mpi baselines/ppo2 baselines/a2c baselines/deepq baselines/acer + - flake8 . - docker run baselines-test pytest -v . diff --git a/baselines/run.py b/baselines/run.py index 15424b9..6556049 100644 --- a/baselines/run.py +++ b/baselines/run.py @@ -1,6 +1,5 @@ import sys import multiprocessing -import os import os.path as osp import gym from collections import defaultdict @@ -9,7 +8,7 @@ import numpy as np from baselines.common.vec_env.vec_frame_stack import VecFrameStack from baselines.common.cmd_util import common_arg_parser, parse_unknown_args, make_mujoco_env, make_atari_env -from baselines.common.tf_util import save_state, load_state, get_session +from baselines.common.tf_util import get_session from baselines import bench, logger from importlib import import_module diff --git a/setup.cfg b/setup.cfg new file mode 100644 index 0000000..2ca999f --- /dev/null +++ b/setup.cfg @@ -0,0 +1,15 @@ +[flake8] +select = F,E999 +exclude = + .git, + __pycache__, + baselines/her, + baselines/ddpg, + baselines/ppo1, + baselines/bench, + baselines/acktr, + + + + + diff --git a/setup.py b/setup.py index 431b062..d4a00c1 100644 --- a/setup.py +++ b/setup.py @@ -1,5 +1,4 @@ from setuptools import setup, find_packages -from functools import reduce import sys if sys.version_info.major != 3: