From 97403ab431aa70cee8350ca20d1a3c70e86ea345 Mon Sep 17 00:00:00 2001 From: kris Date: Wed, 25 Jan 2023 22:57:18 +0000 Subject: [PATCH] Create data dir before writing to it --- transcoder/make_data_tables.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/transcoder/make_data_tables.py b/transcoder/make_data_tables.py index baf464b..66260a1 100644 --- a/transcoder/make_data_tables.py +++ b/transcoder/make_data_tables.py @@ -1,6 +1,5 @@ -import bz2 import functools -import pickle +import os import sys from typing import Iterable, Type @@ -17,7 +16,7 @@ import screen PIXEL_CHARS = "0123456789ABCDEF" - +DATA_DIR = "transcoder/data" def pixel_char(i: int) -> str: return PIXEL_CHARS[i] @@ -184,12 +183,17 @@ def make_edit_distance( """Write file containing (D)HGR edit distance matrix for a palette.""" dist = compute_edit_distance(edp, bitmap_cls, nominal_colours) - data = "transcoder/data/%s_palette_%d_edit_distance.npz" % ( - bitmap_cls.NAME, pal.ID.value) + data = "%s/%s_palette_%d_edit_distance.npz" % ( + DATA_DIR, bitmap_cls.NAME, pal.ID.value) np.savez_compressed(data, edit_distance=dist) def main(): + try: + os.mkdir(DATA_DIR, mode=0o755) + except FileExistsError: + pass + for p in palette.PALETTES.values(): print("Processing palette %s" % p) edp = compute_substitute_costs(p)