ruff

deepsense-ai · Nov 29, 2024 · e93a656 · e93a656
1 parent a9f47d4
commit e93a656
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/packages/ragbits-core/src/ragbits/core/utils/dict_transformations.py b/packages/ragbits-core/src/ragbits/core/utils/dict_transformations.py
@@ -3,9 +3,7 @@
 SimpleTypes = str | int | float | bool | None
 
 
-def flatten_dict(
-    input_dict: dict[str, Any], parent_key: str = "", sep: str = "."
-) -> dict[str, SimpleTypes]:
+def flatten_dict(input_dict: dict[str, Any], parent_key: str = "", sep: str = ".") -> dict[str, SimpleTypes]:
     """
     Recursively flatten a nested dictionary and lists, converting non-primitive types to strings.
 

diff --git a/packages/ragbits-core/src/ragbits/core/vector_stores/chroma.py b/packages/ragbits-core/src/ragbits/core/vector_stores/chroma.py
@@ -184,4 +184,4 @@ async def list(
     @staticmethod
     def _flatten_metadata(metadata: dict) -> dict:
         """Flattens the metadata dictionary. Removes any None values as they are not supported by ChromaDB."""
-        return {k: v for k, v in flatten_dict(metadata).items() if v is not None}
+        return {k: v for k, v in flatten_dict(metadata).items() if v is not None}