Switching to RachisWarning

Author: Macabe222

q2cli/commands.py

@@ -14,7 +14,7 @@
 
 from q2cli.click.command import BaseCommandMixin
 from q2cli.core.config import CONFIG
-from q2cli.util import filter_QIIME2_warnings
+from q2cli.util import filter_rachis_warnings
 
 
 class RootCommand(BaseCommandMixin, click.MultiCommand):
@@ -549,7 +549,7 @@ def __call__(self, **kwargs):
         cleanup_logfile = False
         try:
             with (qiime2.util.redirected_stdio(stdout=log, stderr=log),
-                 filter_QIIME2_warnings()):
+                 filter_rachis_warnings()):
                 if parallel:
                     from qiime2.sdk.parallel_config import \
                         (load_config_from_file, ParallelConfig)

q2cli/util.py

@@ -536,10 +536,10 @@ def get_default_recycle_pool(plugin_action):
 
 
 @contextlib.contextmanager
-def filter_QIIME2_warnings():
+def filter_rachis_warnings():
     import sys
     import warnings
-    from qiime2.core.exceptions import QIIME2Warning
+    from qiime2.core.exceptions import RachisWarning
 
     original_showwarning = warnings.showwarning
 
@@ -553,7 +553,7 @@ def save_warnings(
         warnings of category QIIME2Warning.
         """
 
-        if issubclass(category, QIIME2Warning):
+        if issubclass(category, RachisWarning):
             captured_warnings.append((str(message), filename, lineno))
         else:
             original_showwarning(