[DEVX-454]: PR Review Code Changes

Author: srikanthbachala20

clarifai_datautils/base/features.py

@@ -45,7 +45,7 @@ class VisualSegmentationFeatures:
 class TextFeatures:
   """Text classification datasets preprocessing output features."""
   text: str
-  labels: List[Union[str, int]]  # List[str or int] to cater for multi-class tasks
+  labels: List[Union[str, int]] = None  # List[str or int] to cater for multi-class tasks
   id: Optional[int] = None  # text_id
   metadata: Optional[dict] = None
   label_ids: Optional[List[str]] = None

clarifai_datautils/multimodal/pipeline/loaders.py

@@ -39,8 +39,7 @@ def __getitem__(self, index: int):
     if self.elements[index].to_dict()['type'] == 'Table':
       meta['type'] = 'table'
 
-    return MultiModalFeatures(
-        text=text, image_bytes=image_data, labels=[self.pipeline_name], metadata=meta, id=id)
+    return MultiModalFeatures(text=text, image_bytes=image_data, metadata=meta, id=id)
 
   def __len__(self):
     return len(self.elements)
@@ -65,10 +64,7 @@ def __getitem__(self, index: int):
     id = self.elements[index].to_dict().get('element_id', None)
     id = id[:48] if id is not None else None
     return TextFeatures(
-        text=self.elements[index].text,
-        labels=self.pipeline_name,
-        metadata=self.elements[index].metadata.to_dict(),
-        id=id)
+        text=self.elements[index].text, metadata=self.elements[index].metadata.to_dict(), id=id)
 
   def __len__(self):
     return len(self.elements)