remove omegaconf

deeppavlov · Jan 16, 2025 · 7f04bef · 7f04bef
1 parent 683538f
commit 7f04bef
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/autointent/context/_utils.py b/autointent/context/_utils.py
@@ -9,22 +9,21 @@
 from typing import Any
 
 import numpy as np
-from omegaconf import ListConfig
 
 from autointent import Dataset
 
 
 class NumpyEncoder(json.JSONEncoder):
     """
-    JSON encoder that handles numpy data types and OmegaConf ListConfig.
+    JSON encoder that handles numpy data types.
 
     This encoder extends the default `json.JSONEncoder` to serialize numpy
-    arrays, numpy data types, and OmegaConf ListConfig objects.
+    arrays, numpy data types.
     """
 
     def default(self, obj: Any) -> str | int | float | list[Any] | Any:  # noqa: ANN401
         """
-        Serialize objects with special handling for numpy and OmegaConf types.
+        Serialize objects with special handling for numpy.
 
         :param obj: Object to serialize.
         :return: JSON-serializable representation of the object.
@@ -35,8 +34,6 @@ def default(self, obj: Any) -> str | int | float | list[Any] | Any:  # noqa: ANN
             return float(obj)
         if isinstance(obj, np.ndarray):
             return obj.tolist()
-        if isinstance(obj, ListConfig):
-            return list(obj)
         return super().default(obj)
 
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -181,7 +181,6 @@ module = [
     "appdirs",
     "sre_yield",
     "skmultilearn.model_selection",
-    "omegaconf",
     "hydra",
     "hydra.*",
     "transformers",