diff --git a/setup.py b/setup.py index ff58d7e..3c48ce3 100644 --- a/setup.py +++ b/setup.py @@ -6,7 +6,12 @@ # file, You can obtain one at https://mozilla.org/MPL/2.0/. """Lithium setup script""" +import site + from setuptools import setup +site.ENABLE_USER_SITE = True + + if __name__ == "__main__": setup() diff --git a/src/lithium/reducer.py b/src/lithium/reducer.py index 6782a30..ce94576 100644 --- a/src/lithium/reducer.py +++ b/src/lithium/reducer.py @@ -171,10 +171,10 @@ def error(_, message): early_parser.add_argument( "--strategy", default=DEFAULT_STRATEGY, choices=strategies.keys() ) - args = early_parser.parse_known_args(argv)[0] - atom = args.atom if args else DEFAULT_TESTCASE + args = early_parser.parse_known_args(argv) + atom = args[0].atom if args else DEFAULT_TESTCASE self.strategy = strategies.get( - args.strategy if args else None, strategies[DEFAULT_STRATEGY] + args[0].strategy if args else None, strategies[DEFAULT_STRATEGY] )() grp_opt.add_argument(