Fix issue #93 (#96)

* Fix issue #93 * Add tests * Allow for white space and remove comments after `auto-ignore` * Update test
google-research · Mar 26, 2024 · 3859052 · 3859052
1 parent b7a0b3c
commit 3859052
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/arxiv_latex_cleaner/arxiv_latex_cleaner.py b/arxiv_latex_cleaner/arxiv_latex_cleaner.py
@@ -212,8 +212,10 @@ def _remove_iffalse_block(text):
 
 def _remove_comments_inline(text):
   """Removes the comments from the string 'text' and ignores % inside \\url{}."""
-  if 'auto-ignore' in text:
-    return text
+  auto_ignore_pattern = r'(%\s*auto-ignore).*'
+  if regex.search(auto_ignore_pattern, text):
+    return regex.sub(auto_ignore_pattern, r'\1', text)
+
   if text.lstrip(' ').lstrip('\t').startswith('%'):
     return ''
 

diff --git a/arxiv_latex_cleaner/tests/arxiv_latex_cleaner_test.py b/arxiv_latex_cleaner/tests/arxiv_latex_cleaner_test.py
@@ -278,6 +278,16 @@ def test_merge_args_into_config(self, args, config_params, final_args):
           'line_in': '%auto-ignore\n',
           'true_output': '%auto-ignore\n',
       },
+      {
+          'testcase_name': 'auto_ignore_middle',
+          'line_in': 'Foo % auto-ignore Comment\n',
+          'true_output': 'Foo % auto-ignore\n',
+      },
+      {
+          'testcase_name': 'auto_ignore_text_with_comment',
+          'line_in': 'Foo auto-ignore % Comment\n',
+          'true_output': 'Foo auto-ignore %\n',
+      },
       {
           'testcase_name': 'percent',
           'line_in': r'100\% accurate\n',