Supress userwarning: label 'included' not exported (asreview#24)

Co-authored-by: Jonathan de Bruin <jonathandebruinos@gmail.com>
laurens88 · Nov 18, 2022 · c2b5272 · c2b5272
1 parent 9e499e4
commit c2b5272
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/asreviewcontrib/datatools/stack.py b/asreviewcontrib/datatools/stack.py
@@ -1,4 +1,5 @@
 import argparse
+import warnings
 from pathlib import Path
 
 import pandas as pd
@@ -30,7 +31,10 @@ def vstack(output_file, input_files):
     df_vstacked = pd.concat(list_dfs).reset_index(drop=True)
     as_vstacked = ASReviewData(df=df_vstacked)
 
-    as_vstacked.to_file(output_file)
+    # supress warning about certain columns not exported to .ris output
+    with warnings.catch_warnings():
+        warnings.filterwarnings("ignore", category=UserWarning)
+        as_vstacked.to_file(output_file)
 
 
 def _parse_arguments_vstack():