diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-10-06 10:55:21 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-06 15:09:49 +0000 |
| commit | efa61d31682fb589956cbffa60243d68d9a0daa7 (patch) | |
| tree | 63b0b2fc374020b3c673d5b810d3f8678e7b3f04 /scripts/xy_grid.py | |
| parent | 5d0e6ab8567bda2ee8f5ed31f332ca07c1b84b98 (diff) | |
| download | stable-diffusion-webui-gfx803-efa61d31682fb589956cbffa60243d68d9a0daa7.tar.gz stable-diffusion-webui-gfx803-efa61d31682fb589956cbffa60243d68d9a0daa7.tar.bz2 stable-diffusion-webui-gfx803-efa61d31682fb589956cbffa60243d68d9a0daa7.zip | |
use csv.reader
Diffstat (limited to 'scripts/xy_grid.py')
| -rw-r--r-- | scripts/xy_grid.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/scripts/xy_grid.py b/scripts/xy_grid.py index 210829a7..1a625898 100644 --- a/scripts/xy_grid.py +++ b/scripts/xy_grid.py @@ -1,8 +1,9 @@ from collections import namedtuple
from copy import copy
-from itertools import permutations
+from itertools import permutations, chain
import random
-
+import csv
+from io import StringIO
from PIL import Image
import numpy as np
@@ -168,8 +169,6 @@ re_range_float = re.compile(r"\s*([+-]?\s*\d+(?:.\d*)?)\s*-\s*([+-]?\s*\d+(?:.\d re_range_count = re.compile(r"\s*([+-]?\s*\d+)\s*-\s*([+-]?\s*\d+)(?:\s*\[(\d+)\s*\])?\s*")
re_range_count_float = re.compile(r"\s*([+-]?\s*\d+(?:.\d*)?)\s*-\s*([+-]?\s*\d+(?:.\d*)?)(?:\s*\[(\d+(?:.\d*)?)\s*\])?\s*")
-re_non_escaped_comma = re.compile(r"(?<!\\),")
-
class Script(scripts.Script):
def title(self):
return "X/Y plot"
@@ -198,7 +197,7 @@ class Script(scripts.Script): if opt.label == 'Nothing':
return [0]
- valslist = [x.replace('\\,',',').strip() for x in re_non_escaped_comma.split(vals)]
+ valslist = list(chain.from_iterable(csv.reader(StringIO(s))))
if opt.type == int:
valslist_ext = []
|