diff --git a/chroma_feedback/consumer/core.py b/chroma_feedback/consumer/core.py index 85bd0e4..6393992 100644 --- a/chroma_feedback/consumer/core.py +++ b/chroma_feedback/consumer/core.py @@ -31,6 +31,6 @@ def process(program : ArgumentParser, producer_report : List[ProducerReport]) -> def load_consumer(consumer_name : str) -> Any: try: return importlib.import_module(consumer.ALL[consumer_name]) - except ImportError: + except ModuleNotFoundError: logger.error(wording.get('consumer_not_found').format(consumer.ALL[consumer_name]) + wording.get('exclamation_mark')) sys.exit() diff --git a/chroma_feedback/producer/core.py b/chroma_feedback/producer/core.py index c114b93..c793d25 100644 --- a/chroma_feedback/producer/core.py +++ b/chroma_feedback/producer/core.py @@ -27,6 +27,6 @@ def process(program : ArgumentParser) -> List[Producer]: def load_producer(producer_name : str) -> Any: try: return importlib.import_module(producer.ALL[producer_name]) - except ImportError: + except ModuleNotFoundError: logger.error(wording.get('producer_not_found').format(producer.ALL[producer_name]) + wording.get('exclamation_mark')) sys.exit()