From c149e6baea4a44c02489de54db338f52bdc59dd2 Mon Sep 17 00:00:00 2001 From: sepandhaghighi Date: Tue, 28 Sep 2021 15:20:04 +0330 Subject: [PATCH] fix : plot method updated #10 --- samila/genimage.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/samila/genimage.py b/samila/genimage.py index b374905..d8585df 100644 --- a/samila/genimage.py +++ b/samila/genimage.py @@ -3,7 +3,7 @@ import random import itertools import matplotlib.pyplot as plt -from .functions import float_range, filter_color +from .functions import float_range, filter_color, filter_projection from .params import * @@ -70,6 +70,7 @@ class GenerativeImage: color = filter_color(color) if not None else DEFAULT_COLOR bgcolor = filter_color( bgcolor) if not None else DEFAULT_BACKGROUND_COLOR + projection = filter_projection(projection) fig = plt.figure() fig.set_size_inches(size[0], size[1]) fig.set_facecolor(bgcolor)