From 5ee0e400090ebccedbc4f7855513766f92a0698d Mon Sep 17 00:00:00 2001
From: marshallmallows <yuanzhu.zhan@graviti.com>
Date: Thu, 18 Nov 2021 16:04:29 +0800
Subject: [PATCH] refactor(opendataset): remove relucdant attributes of BDD100K
 dataset

remove the "areaType" of "POLYGON" and the "laneTypes" of "POLYLINE2D"
---
 .../opendataset/BDD100K/catalog_100k.json     | 21 +------------------
 tensorbay/opendataset/BDD100K/loader.py       |  5 +++--
 2 files changed, 4 insertions(+), 22 deletions(-)

diff --git a/tensorbay/opendataset/BDD100K/catalog_100k.json b/tensorbay/opendataset/BDD100K/catalog_100k.json
index 065f4d247..969047979 100644
--- a/tensorbay/opendataset/BDD100K/catalog_100k.json
+++ b/tensorbay/opendataset/BDD100K/catalog_100k.json
@@ -64,13 +64,7 @@
         ]
     },
     "POLYGON": {
-        "categories": [{ "name": "direct" }, { "name": "alternative" }, { "name": "unknown" }],
-        "attributes": [
-            {
-                "name": "areaType",
-                "enum": ["direct", "alternative"]
-            }
-        ]
+        "categories": [{ "name": "direct" }, { "name": "alternative" }, { "name": "unknown" }]
     },
     "POLYLINE2D": {
         "isBeizerCurve": true,
@@ -92,19 +86,6 @@
             {
                 "name": "laneStyle",
                 "enum": ["solid", "dashed"]
-            },
-            {
-                "name": "laneTypes",
-                "enum": [
-                    "crosswalk",
-                    "double other",
-                    "double white",
-                    "double yellow",
-                    "road curb",
-                    "single other",
-                    "single white",
-                    "single yellow"
-                ]
             }
         ]
     }
diff --git a/tensorbay/opendataset/BDD100K/loader.py b/tensorbay/opendataset/BDD100K/loader.py
index e12e80501..1975fbd71 100644
--- a/tensorbay/opendataset/BDD100K/loader.py
+++ b/tensorbay/opendataset/BDD100K/loader.py
@@ -265,14 +265,15 @@ def _add_poly2d_label_100k(
         labeled_polygon = LabeledPolygon(
             points=poly2d_info["vertices"],
             category=label_info["category"],
-            attributes=label_info["attributes"],
         )
         polygon.append(labeled_polygon)
     else:
+        attributes = label_info["attributes"]
+        attributes.pop("laneTypes", None)
         labeled_polyline2d = LabeledPolyline2D(
             points=poly2d_info["vertices"],
             category=label_info["category"],
-            attributes=label_info.get("attributes", {}),
+            attributes=attributes,
             beizer_point_types=poly2d_info["types"],
         )
         polyline2d.append(labeled_polyline2d)