Spaces:

albertvillanova
/

datasets-tagging

Runtime error

App Files Files Community

Thomas Wolf commited on Dec 2, 2020

Commit

49695dd

•

2 Parent(s): 97e23ac 7edcd52

Merge pull request #6 from patil-suraj/handle-list-feature

Browse files

Files changed (1) hide show

tagging_app.py +22 -20

tagging_app.py CHANGED Viewed

@@ -57,43 +57,45 @@ creator_set = {
 ########################
 @st.cache
-def filter_features(feature_dict):
-    if feature_dict.get("_type", None) == 'Value':
         return {
-            "feature_type": feature_dict["_type"],
-            "dtype": feature_dict["dtype"],
         }
-    elif feature_dict.get("_type", None) == 'Sequence':
-        if "dtype" in feature_dict["feature"]:
             return {
-                "feature_type": feature_dict["_type"],
-                "feature": filter_features(feature_dict["feature"]),
             }
-        elif "_type" in feature_dict["feature"] and feature_dict["feature"]["_type"] == "ClassLabel":
             return {
-                "feature_type": feature_dict["_type"],
                 "dtype": "int32",
-                "feature": filter_features(feature_dict["feature"]),
             }
         else:
             return dict(
-                [("feature_type", feature_dict["_type"])] + \
-                [(k, filter_features(v)) for k, v in feature_dict["feature"].items()]
             )
-    elif feature_dict.get("_type", None) == 'ClassLabel':
         return {
-            "feature_type": feature_dict["_type"],
             "dtype": "int32",
-            "class_names": feature_dict["names"],
         }
-    elif feature_dict.get("_type", None) in ['Translation', 'TranslationVariableLanguages']:
         return {
-            "feature_type": feature_dict["_type"],
             "dtype": "string",
-            "languages": feature_dict["languages"],
         }
     else:
-        return dict([(k, filter_features(v)) for k, v in feature_dict.items()])
 @st.cache

 ########################
 @st.cache
+def filter_features(features):
+    if isinstance(features, list):
+        return dict([(k, filter_features(v)) for k, v in features[0].items()])
+    elif features.get("_type", None) == 'Value':
         return {
+            "feature_type": features["_type"],
+            "dtype": features["dtype"],
         }
+    elif features.get("_type", None) == 'Sequence':
+        if "dtype" in features["feature"]:
             return {
+                "feature_type": features["_type"],
+                "feature": filter_features(features["feature"]),
             }
+        elif "_type" in features["feature"] and features["feature"]["_type"] == "ClassLabel":
             return {
+                "feature_type": features["_type"],
                 "dtype": "int32",
+                "feature": filter_features(features["feature"]),
             }
         else:
             return dict(
+                [("feature_type", features["_type"])] + \
+                [(k, filter_features(v)) for k, v in features["feature"].items()]
             )
+    elif features.get("_type", None) == 'ClassLabel':
         return {
+            "feature_type": features["_type"],
             "dtype": "int32",
+            "class_names": features["names"],
         }
+    elif features.get("_type", None) in ['Translation', 'TranslationVariableLanguages']:
         return {
+            "feature_type": features["_type"],
             "dtype": "string",
+            "languages": features["languages"],
         }
     else:
+        return dict([(k, filter_features(v)) for k, v in features.items()])
 @st.cache