Fix tracemalloc error when action isn't defined (#161)

psf · Jun 11, 2023 · f4db461 · f4db461
1 parent 8dc692f
commit f4db461
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/doc/changelog.rst b/doc/changelog.rst
@@ -1,6 +1,12 @@
 Changelog
 =========
 
+Version 2.6.1
+-------------
+
+* Fix a possible attribute error in argument parsing.
+  Patch by Colton Myers
+
 Version 2.6.0 (2023-03-22)
 --------------------------
 

diff --git a/pyperf/_runner.py b/pyperf/_runner.py
@@ -331,7 +331,7 @@ def _process_args_impl(self):
             self._only_in_worker("--worker-task")
 
         if args.tracemalloc:
-            if args.action == 'command':
+            if getattr(args, 'action', None) == 'command':
                 raise CLIError('--tracemalloc cannot be used with pyperf command')
             try:
                 import tracemalloc   # noqa