From de8b4b317b903d046161117010865ec4ea843d19 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Maciej=20P=C4=99=C5=9Bko?= Date: Sun, 18 Nov 2018 01:34:43 +0100 Subject: [PATCH] Fix minor bug --- api/models.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/api/models.py b/api/models.py index b377183..2f940a0 100644 --- a/api/models.py +++ b/api/models.py @@ -75,7 +75,8 @@ class Comic(models.Model): @classmethod @profile - def create_from_nparray(cls, nparray, video, yt_url, frames_mode, rl_mode, image_assess_mode, style_transfer_mode): + def create_from_nparray(cls, nparray, video, yt_url, frames_mode, + rl_mode, image_assessment_mode, style_transfer_mode): # if nparray.max() <= 1: # nparray = nparray.astype(int) tmp_name = uuid.uuid4().hex + ".png" @@ -87,7 +88,7 @@ class Comic(models.Model): yt_url=yt_url, frames_mode=frames_mode, rl_mode=rl_mode, - image_assessment_mode=image_assess_mode, + image_assessment_mode=image_assessment_mode, style_transfer_mode=style_transfer_mode) os.remove(jj(settings.TMP_DIR, tmp_name)) return comic