Merge pull request #11 from Project-AgML/dev

Release 0.2.6

Author: amogh7joshi

agml/_assets/public_datasources.json

@@ -76,7 +76,7 @@
         "platform": "ground_fixed",
         "input_data_format": "png",
         "annotation_format": "directory_names",
-        "n_images": "5588",
+        "n_images": "5539",
         "docs_url": "https://vision.eng.au.dk/plant-seedlings-dataset/",
         "stats": {
             "mean": [
@@ -381,38 +381,6 @@
         },
         "external_image_sources": []
     },
-    "cotton_seedling_counting": {
-        "ml_task": "object_detection",
-        "ag_task": "seedling_counting",
-        "location": {
-            "continent": "worldwide",
-            "country": "worldwide"
-        },
-        "sensor_modality": "rgb",
-        "real_synthetic": "real",
-        "platform": "ground",
-        "input_data_format": "png",
-        "annotation_format": "coco_json",
-        "n_images": "2290",
-        "docs_url": "https://github.com/UGA-BSAIL/deepseedling",
-        "stats": {
-            "mean": [
-                0.6695063710212708,
-                0.5673825144767761,
-                0.4538835883140564
-            ],
-            "std": [
-                0.07037307322025299,
-                0.07951230555772781,
-                0.07661264389753342
-            ]
-        },
-        "classes": {
-            "0": "Plant",
-            "1": "Weed"
-        },
-        "external_image_sources": []
-    },
     "mango_detection_australia": {
         "ml_task": "object_detection",
         "ag_task": "fruit_detection",

agml/data/metadata.py

@@ -25,6 +25,24 @@
 )
 
 
+class _MetadataDict(dict):
+    """Dictionary subclass that throws a custom error for metadata accesses."""
+    def __init__(self, *args, dataset = None, **kwargs):
+        if dataset is None:
+            raise ValueError(
+                "Cannot instantiate metadata dictionary without the dataset name.")
+        super(_MetadataDict, self).__init__(*args, **kwargs)
+        self._dataset = dataset
+
+    def __getitem__(self, item):
+        try:
+            return super(_MetadataDict, self).__getitem__(item)
+        except KeyError:
+            raise KeyError(
+                f"The dataset '{self._dataset}' is missing metadata '{item}'. "
+                f"Please bring this issue to the attention of the AgML team.")
+
+
 class DatasetMetadata(AgMLSerializable):
     """Stores metadata about a certain AgML dataset.
 
@@ -88,10 +106,12 @@ def _load_source_info(self, name):
                     f"Interpreted dataset '{name}' as '{name.replace('-', '_')}.'")
                 name = name.replace('-', '_')
         self._name = name
-        self._metadata = source_info[name]
+        self._metadata = _MetadataDict(
+            **source_info[name], dataset = name)
 
         # Load citation information.
-        self._citation_meta = load_citation_sources()[name]
+        self._citation_meta = _MetadataDict(
+            **load_citation_sources()[name], dataset = name)
 
     def __getattr__(self, key):
         try: