debug parser's skeleton

opendatalab · Apr 21, 2023 · 8f69d6b · 8f69d6b
1 parent 091f695
commit 8f69d6b
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/dsdl/parser/parse_class.py b/dsdl/parser/parse_class.py
@@ -31,7 +31,7 @@ def _parse_class_name(raw_name: str) -> str:
         return raw_name
 
     def _parse_skeleton(self, skeleton: _List) -> Optional[_List[_List[int]]]:
-        if not skeleton:
+        if skeleton is None:
             return None
         else:
             if isinstance(skeleton, list):

diff --git a/dsdl/parser/parser.py b/dsdl/parser/parser.py
@@ -438,7 +438,7 @@ def _generate(self) -> Optional[str]:
                 for ele_class in val.field_list:
                     dsdl_py += f""""{ele_class.label_value}","""
                 dsdl_py += "],\n"
-                if val.skeleton:
+                if val.skeleton is not None:
                     dsdl_py += f"""    skeleton = {val.skeleton}\n"""
                 dsdl_py += ")\n"
             if idx != len(ordered_keys) - 1:

diff --git a/dsdl/tools/view.py b/dsdl/tools/view.py
@@ -38,7 +38,6 @@ def view(dsdl_yaml, num, random, visualize, fields, config, position, multistage
         else:
             dsdl_py = dsdl_parse(dsdl_yaml["yaml_file"], dsdl_library_path="")
         exec(dsdl_py, {})
-
     dataset = Dataset(dsdl_yaml["samples"], dsdl_yaml["sample_type"], config,
                       global_info=dsdl_yaml["global_info"], global_info_type=dsdl_yaml["global_info_type"])