Fixing flake8 issue and adding kwargs for plotting
This MR fixes an issue with flake8
which is related to #110 (closed) taking long to run the pre-commit hook.
In addition kwargs are introduced in the input variable plotting
This MR fixes an issue with flake8
which is related to #110 (closed) taking long to run the pre-commit hook.
In addition kwargs are introduced in the input variable plotting