From dd39e5d94490d89b2617d154509e3bceb1cc4aa2 Mon Sep 17 00:00:00 2001
From: Kohya S <ykumeykume@gmail.com>
Date: Sun, 5 Mar 2023 20:04:18 +0900
Subject: [PATCH] hope to support python 3.8/3.9

---
 library/config_util.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/library/config_util.py b/library/config_util.py
index a253bfb1..afc9be75 100644
--- a/library/config_util.py
+++ b/library/config_util.py
@@ -3,10 +3,11 @@ from dataclasses import (
   asdict,
   dataclass,
 )
+import functools
 from textwrap import dedent, indent
 import json
 from pathlib import Path
-from toolz import curry
+# from toolz import curry
 from typing import (
   List,
   Optional,
@@ -111,13 +112,13 @@ class Blueprint:
 
 
 class ConfigSanitizer:
-  @curry
+  # @curry
   @staticmethod
   def __validate_and_convert_twodim(klass, value: Sequence) -> Tuple:
     Schema(ExactSequence([klass, klass]))(value)
     return tuple(value)
 
-  @curry
+  # @curry
   @staticmethod
   def __validate_and_convert_scalar_or_twodim(klass, value: Union[float, Sequence]) -> Tuple:
     Schema(Any(klass, ExactSequence([klass, klass])))(value)
@@ -130,7 +131,7 @@ class ConfigSanitizer:
   # subset schema
   SUBSET_ASCENDABLE_SCHEMA = {
     "color_aug": bool,
-    "face_crop_aug_range": __validate_and_convert_twodim(float),
+    "face_crop_aug_range": functools.partial(__validate_and_convert_twodim.__func__, float),
     "flip_aug": bool,
     "num_repeats": int,
     "random_crop": bool,
@@ -166,7 +167,7 @@ class ConfigSanitizer:
     "enable_bucket": bool,
     "max_bucket_reso": int,
     "min_bucket_reso": int,
-    "resolution": __validate_and_convert_scalar_or_twodim(int),
+    "resolution": functools.partial(__validate_and_convert_scalar_or_twodim, int),
   }
 
   # options handled by argparse but not handled by user config