tensorneko-util 0.3.15__py3-none-any.whl → 0.3.16__py3-none-any.whl

tensorneko_util/io/json/json_writer.py

@@ -14,7 +14,7 @@ class JsonWriter:
                 json.dump(obj, file, indent=indent, ensure_ascii=ensure_ascii)
         else:
             import orjson
-            with open(path, "w", encoding=encoding) as file:
+            with open(path, "wb") as file:
                 if indent == 4:
                     warnings.warn("orjson does not support indent 4, will use indent 2 instead.")
                     option = orjson.OPT_INDENT_2

tensorneko_util/util/registry.py

@@ -28,7 +28,7 @@ class Registry(Generic[T]):
     """
 
     def __init__(self):
-        self._registry: dict[str, T] = {}
+        self._registry: dict[str, Type[T]] = {}
 
     def register(self, name: str) -> Callable[[Type[T]], Type[T]]:
         def wrapper(cls):
@@ -37,7 +37,7 @@ class Registry(Generic[T]):
 
         return wrapper
 
-    def __getitem__(self, name: str) -> T:
+    def __getitem__(self, name: str) -> Type[T]:
         return self._registry[name]
 
     def items(self):

tensorneko_util/version.txt

@@ -1 +1 @@
-0.3.15
+0.3.16