Be robust to some argparse string variants

RfidResearchGroup · Nov 6, 2023 · 4d670f0 · 4d670f0
1 parent ecf3c06
commit 4d670f0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/software/script/chameleon_utils.py b/software/script/chameleon_utils.py
@@ -88,8 +88,10 @@ def print_help(self):
                 options = lines
                 lines = []
         if len(options) > 0:
-            assert options[0].strip() == 'options:'
+            # 2 variants depending on Python version(?)
+            assert options[0].strip() in ['options:', 'optional arguments:']
             options[0] = options[0].replace('options:', f'{CG}options:{C0}')
+            options[0] = options[0].replace('optional arguments:', f'{CG}optional arguments:{C0}')
             if len(options) > 1:
                 options.append('')
             print('\n'.join(options))