Update readme index weaviate component (#843)

Small changes to the index weaviate component readme.
ml6team · Feb 6, 2024 · 7ec04dc · 7ec04dc
1 parent a0ba72c
commit 7ec04dc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/components/index_weaviate/README.md b/components/index_weaviate/README.md
@@ -24,7 +24,7 @@ dataset = pipeline.read(
     }
 )
 
-dataset = dataset.apply(
+dataset.write(
     "index_weaviate",
     arguments={
         "weaviate_url": "http://localhost:8080",
@@ -35,7 +35,7 @@ dataset = dataset.apply(
         }
     },
     consumes={
-        "text": "text"
+        "text": pa.string()
     }
 )
 ```
@@ -66,14 +66,14 @@ dataset = dataset.apply(
     },
 )
 
-dataset = dataset.apply(
+dataset.write(
     "index_weaviate",
     arguments={
         "weaviate_url": "http://localhost:8080",
         "class_name": "my_class",
     },
     consumes={
-        "embedding": "embedding"
+        "embedding": pa.list_(pa.float32())
     }
 )
 ```

diff --git a/components/index_weaviate/fondant_component.yaml b/components/index_weaviate/fondant_component.yaml
@@ -22,7 +22,7 @@ description: |
       }
   )
 
-  dataset = dataset.apply(
+  dataset.write(
       "index_weaviate",
       arguments={
           "weaviate_url": "http://localhost:8080",
@@ -33,7 +33,7 @@ description: |
           }
       },
       consumes={
-          "text": "text"
+          "text": pa.string()
       }
   )
   ```
@@ -64,14 +64,14 @@ description: |
       },
   )
   
-  dataset = dataset.apply(
+  dataset.write(
       "index_weaviate",
       arguments={
           "weaviate_url": "http://localhost:8080",
           "class_name": "my_class",
       },
       consumes={
-          "embedding": "embedding"
+          "embedding": pa.list_(pa.float32())
       }
   )
   ```