remove ujson

zeptofine · Oct 9, 2023 · 2a0caa8 · 2a0caa8
1 parent aef7a2a
commit 2a0caa8
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 80 deletions.
diff --git a/imdataset_creator/datarules/dataset_builder.py b/imdataset_creator/datarules/dataset_builder.py
@@ -6,7 +6,6 @@
 from typing import Generator, Literal, TypeVar, overload
 
 import polars as pl
-import ujson
 from polars import DataFrame, Expr, concat
 from polars.type_aliases import SchemaDefinition
 

diff --git a/imdataset_creator/main.py b/imdataset_creator/main.py
@@ -1,5 +1,6 @@
 # from __future__ import annotations
 
+import json
 import logging
 import os
 from collections.abc import Callable, Iterable
@@ -13,7 +14,6 @@
 import numpy as np
 import rich.progress as progress
 import typer
-import ujson
 from polars import DataFrame, concat
 from rich.console import Console
 from rich.logging import RichHandler
@@ -22,17 +22,17 @@
 from typing_extensions import Annotated
 
 from . import (
-    MainConfig,
+    ConfigHandler,
+    DatasetBuilder,
     File,
+    FileScenario,
     Input,
+    MainConfig,
     Output,
+    OutputScenario,
     Producer,
     Rule,
-    ConfigHandler,
-    DatasetBuilder,
     chunk_split,
-    FileScenario,
-    OutputScenario,
 )
 
 CPU_COUNT = int(cpu_count())
@@ -84,7 +84,7 @@ def main(
         return 0
 
     with config_path.open("r") as f:
-        cfg: MainConfig = ujson.load(f)
+        cfg: MainConfig = json.load(f)
 
     db = DatasetBuilder(db_path=Path(database_path))
 

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -22,7 +22,6 @@ python = "^3.10, <3.12"
 python-dateutil = "^2.8.2"
 rich = "^13.5.3"
 typer = {extras = ["all"], version = "^0.9.0"}
-ujson = "^5.8.0"
 wcmatch = "^8.5"
 
 [tool.poetry.scripts]