Merge pull request #2054 from tk0miya/bugs/UntypedFunction.map_type_name

Fix UntypedFunction does not provide #map_type_name
ruby · Oct 8, 2024 · a877d7d · a877d7d
2 parents fc3c07f + 00c7c52
commit a877d7d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lib/rbs/types.rb b/lib/rbs/types.rb
@@ -1221,6 +1221,12 @@ def map_type(&block)
         end
       end
 
+      def map_type_name(&block)
+        UntypedFunction.new(
+          return_type: return_type.map_type_name(&block)
+        )
+      end
+
       def each_type(&block)
         if block
           yield return_type