From 858b2664db42339ab0808c538a37a9fde544c0d7 Mon Sep 17 00:00:00 2001 From: kvid Date: Wed, 12 Jun 2024 22:49:09 +0200 Subject: [PATCH] Update src/wireviz/wireviz.py (#383) Raising TypeError is better than assert. (Black reformatted) Co-authored-by: Andreas Motl --- src/wireviz/wireviz.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/wireviz/wireviz.py b/src/wireviz/wireviz.py index ad3babc..6e0ab82 100755 --- a/src/wireviz/wireviz.py +++ b/src/wireviz/wireviz.py @@ -88,9 +88,10 @@ def parse( raise Exception("No output formats or return types specified") yaml_data, yaml_file = _get_yaml_data_and_path(inp) - assert isinstance( - yaml_data, dict - ), f"Expected a dict as top-level YAML input, but got: {type(yaml_data)}" + if not isinstance(yaml_data, dict): + raise TypeError( + f"Expected a dict as top-level YAML input, but got: {type(yaml_data)}" + ) if output_formats: # need to write data to file, determine output directory and filename output_dir = _get_output_dir(yaml_file, output_dir)