Merge pull request #37 from yummyml/rust-v-embeddings

fix backend
yummyml · Jan 28, 2024 · fcac1eb · fcac1eb
2 parents 189e5b3 + 5489eaf
commit fcac1eb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -136,3 +136,4 @@ dmypy.json
 yummy-rs/protos
 yummy-rs/yummy-core/protos
 Cargo.lock
+.benchmark
diff --git a/setup.py b/setup.py
@@ -61,7 +61,7 @@
 
 setup(
     name=NAME,
-    version="0.0.10",
+    version="0.0.11",
     author="Qooba",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",

diff --git a/yummy/backends/backend.py b/yummy/backends/backend.py
@@ -434,7 +434,7 @@ def read_schema(
         elif backend_type in [BackendType.ray, BackendType.dask]:
             return [(k,str(v)) for (k,v) in df.dtypes.to_dict().items()]
         elif backend_type == BackendType.polars:
-            return [(k,v.__name__.lower()) for (k,v) in df.schema.items()]
+            return [(k,str(v).lower()) for (k,v) in df.schema.items()]
 
 
 class YummyOfflineStoreConfig(BackendConfig):