WIP

nutti · Oct 8, 2024 · 8dfdbec · 8dfdbec
1 parent d53ea09
commit 8dfdbec
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/src/fake_bpy_module/transformer/data_type_refiner.py b/src/fake_bpy_module/transformer/data_type_refiner.py
@@ -783,10 +783,8 @@ def _parse_from_description(
  REGEX_MATCH_DESCRIPTION_TYPE_IN.match(description_str):
  enum_literal_type = get_rna_enum_name(description_str)
  dtype_node = DataTypeNode()
- append_child(dtype_node, nodes.Text("set["))
  append_child(dtype_node,
  EnumRef(text=f"bpy.typing.{enum_literal_type}"))
- append_child(dtype_node, nodes.Text("]"))
  return [dtype_node], True
 
  return [], False

diff --git a/...y_module_test/integration_test_data/integration_test/expect/multiple/module_2/__init__.py b/...y_module_test/integration_test_data/integration_test/expect/multiple/module_2/__init__.py
@@ -22,9 +22,9 @@ def function_1(
  :param arg_3: function_1 arg_3 description
  :type arg_3: bpy.typing.Enum1
  :param arg_4: Enumerator in `rna_enum_enum1`.
- :type arg_4: set[bpy.typing.Enum1]
+ :type arg_4: bpy.typing.Enum1
  :param arg_5: job type in `rna_enum_enum1`.
- :type arg_5: set[bpy.typing.Enum1]
+ :type arg_5: bpy.typing.Enum1
  :return: method_1 return description
  :rtype: module_1.submodule_1.BaseClass1
  """
diff --git a/..._module_test/integration_test_data/integration_test/expect/multiple/module_2/__init__.pyi b/..._module_test/integration_test_data/integration_test/expect/multiple/module_2/__init__.pyi
@@ -18,9 +18,9 @@ def function_1(
  :param arg_3: function_1 arg_3 description
  :type arg_3: bpy.typing.Enum1
  :param arg_4: Enumerator in `rna_enum_enum1`.
- :type arg_4: set[bpy.typing.Enum1]
+ :type arg_4: bpy.typing.Enum1
  :param arg_5: job type in `rna_enum_enum1`.
- :type arg_5: set[bpy.typing.Enum1]
+ :type arg_5: bpy.typing.Enum1
  :return: method_1 return description
  :rtype: module_1.submodule_1.BaseClass1
  """