* Add workaround to support pip -e (pypa/pip#7953)

* Fix exception when invalid strategy is specified
jschwartzentruber · Sep 22, 2020 · ab64902 · ab64902
1 parent 240f546
commit ab64902
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git 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
@@ -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(