chore: adapt to the latest changes (#267)

jina-ai · Dec 22, 2021 · c51b044 · c51b044
1 parent c75b670
commit c51b044
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/docs/index.md b/docs/index.md
@@ -121,7 +121,7 @@ Learn more about {term}`embedding model`.
 </div>
 
 +++
-Learn more about {term}`labeled data`.
+Learn more about {term}`labeled dataset`.
 
 
 :::

diff --git a/finetuner/labeler/__init__.py b/finetuner/labeler/__init__.py
@@ -96,7 +96,7 @@ def get_preprocess_fn(self):
 
     f.expose_endpoint('/next')  #: for allowing client to fetch for the next batch
     f.expose_endpoint('/fit')  #: for signaling the backend to fit on the labeled data
-    f.expose_endpoint('/feed')  #: for signaling the backend to fit on the labeled data
+    f.expose_endpoint('/feed')  #: for feeding data to the DataIterator
     f.expose_endpoint(
         '/save'
     )  #: for signaling the backend to save the current state of the model

diff --git a/finetuner/labeler/executor.py b/finetuner/labeler/executor.py
@@ -70,7 +70,7 @@ def embed(self, docs: DocumentArray, parameters: Dict, **kwargs):
             limit=int(parameters.get('topk', 10)),
             exclude_self=True,
         )
-        for d in docs.traverse_flat(['r', 'm']):
+        for d in docs.traverse_flat('r,m'):
             d.pop('blob', 'embedding')
 
     @requests(on='/fit')
@@ -127,7 +127,7 @@ def take_batch(self, parameters: Dict, **kwargs):
 
     @requests(on='/fit')
     def add_fit_data(self, docs: DocumentArray, **kwargs):
-        for d in docs.traverse_flat(['r', 'm']):
+        for d in docs.traverse_flat('r,m'):
             d.content = self._all_data[d.id].content
         self._labeled_dam.extend(docs)
         return self._labeled_dam