Keep MetaData class in get_single_element

Giskard-AI · Aug 8, 2024 · 496a3c6 · 496a3c6
1 parent ba1055f
commit 496a3c6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/examples/landmark_detection/benchmarks/SixDRepNet_benchmark.ipynb b/examples/landmark_detection/benchmarks/SixDRepNet_benchmark.ipynb
@@ -35,9 +35,10 @@
  " meta_predictions[\"yaw\"].append(round(yaw[0], 1))\n",
  " meta_predictions[\"roll\"].append(round(roll[0], 1))\n",
  "\n",
- " meta_ground_truth[\"pitch\"].append(meta[0][\"faceAttributes_headPose_pitch\"])\n",
- " meta_ground_truth[\"yaw\"].append(-meta[0][\"faceAttributes_headPose_yaw\"])\n",
- " meta_ground_truth[\"roll\"].append(meta[0][\"faceAttributes_headPose_roll\"])"
+ " meta = meta[0].data\n",
+ " meta_ground_truth[\"pitch\"].append(meta[\"faceAttributes_headPose_pitch\"])\n",
+ " meta_ground_truth[\"yaw\"].append(-meta[\"faceAttributes_headPose_yaw\"])\n",
+ " meta_ground_truth[\"roll\"].append(meta[\"faceAttributes_headPose_roll\"])"
  ]
  },
  {

diff --git a/giskard_vision/core/dataloaders/base.py b/giskard_vision/core/dataloaders/base.py
@@ -185,7 +185,7 @@ def get_single_element(self, idx) -> TypesBase.single_data:
  TypesBase.single_data: Tuple containing image, labels, and meta information.
  """
  metadata = self.get_meta_with_default(idx)
- return self.get_image(idx), self.get_labels_with_default(idx), metadata.data if metadata is not None else {}
+ return self.get_image(idx), self.get_labels_with_default(idx), metadata if metadata is not None else MetaData({})
 
  def __getitem__(self, idx: int) -> TypesBase.batched_data:
  """