diff --git a/src/eynollah/plot.py b/src/eynollah/plot.py index b01fc04..412ae5a 100644 --- a/src/eynollah/plot.py +++ b/src/eynollah/plot.py @@ -1,5 +1,8 @@ -import matplotlib.pyplot as plt -import matplotlib.patches as mpatches +try: + import matplotlib.pyplot as plt + import matplotlib.patches as mpatches +except ImportError: + plt = mpatches = None import numpy as np import os.path import cv2 diff --git a/src/eynollah/utils/__init__.py b/src/eynollah/utils/__init__.py index 70216e1..29b1273 100644 --- a/src/eynollah/utils/__init__.py +++ b/src/eynollah/utils/__init__.py @@ -1,13 +1,17 @@ +import time import math -import matplotlib.pyplot as plt +try: + import matplotlib.pyplot as plt +except ImportError: + plt = None import numpy as np from shapely import geometry import cv2 import imutils from scipy.signal import find_peaks from scipy.ndimage import gaussian_filter1d -import time + from .is_nan import isNaN from .contour import (contours_in_same_horizon, find_new_features_of_contours,