Fix bandit

Signed-off-by: Adam Li <[email protected]>

pyproject.toml

@@ -83,7 +83,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poe.tasks]
 _flake8 = 'flake8'
-_bandit = 'bandit -r pywhy_graphs'
+_bandit = 'bandit -r pywhy_graphs -c pyproject.toml'
 _black = 'black .'
 _isort = 'isort .'
 _black_check = 'black --check pywhy_graphs examples'
@@ -170,3 +170,6 @@ omit = ['**/__init__.py', '**/tests/**']
 [tool.coverage.report]
 exclude_lines = ['pragma: no cover', 'if __name__ == .__main__.:']
 precision = 2
+
+[tool.bandit]
+exclude_dirs = ["tests"]

pywhy_graphs/export.py

@@ -106,7 +106,8 @@ def to_numpy(causal_graph):
     # ADMGs can have two edges between any 2 nodes
     if type(causal_graph).__name__ == "ADMG":
         # we handle this case separately from the other graphs
-        assert len(graph_map) == 1
+        if len(graph_map) != 1:
+            raise AssertionError("The number of graph maps should be 1...")
 
         # set all bidirected edges with value 10
         bidirected_graph_arr[bidirected_graph_arr != 0] = 10