From 790a98fc980490e4f63ddc7b54368e3883b0ed78 Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Fri, 16 Aug 2019 14:43:03 +0200
Subject: [PATCH] Missing import

- how could this not have been caught by the tests?!
---
 pystencils/plot.py                       | 1 +
 pystencils/simp/assignment_collection.py | 3 +--
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/pystencils/plot.py b/pystencils/plot.py
index de1b1e3..1555b94 100644
--- a/pystencils/plot.py
+++ b/pystencils/plot.py
@@ -3,6 +3,7 @@ This module extends the pyplot module with functions to show scalar and vector f
 simulation coordinate system (y-axis goes up), instead of the "image coordinate system" (y axis goes down) that
 matplotlib normally uses.
 """
+import warnings
 from itertools import cycle
 
 from matplotlib.pyplot import *
diff --git a/pystencils/simp/assignment_collection.py b/pystencils/simp/assignment_collection.py
index dcd8900..fa437a5 100644
--- a/pystencils/simp/assignment_collection.py
+++ b/pystencils/simp/assignment_collection.py
@@ -5,8 +5,7 @@ import sympy as sp
 
 from pystencils.assignment import Assignment
 from pystencils.simp.simplifications import (
-    sort_assignments_topologically,
-    transform_lhs_and_rhs, transform_rhs)
+    sort_assignments_topologically, transform_lhs_and_rhs, transform_rhs)
 from pystencils.sympyextensions import count_operations, fast_subs
 
 
-- 
GitLab