feat(py) None return type does not display returns (#51)

2022-01-29 19:56:14 +01:00
parent 9b7d6e5e81
commit 0d381d03d2
1 changed files with 3 additions and 0 deletions
--- a/lua/neogen/configurations/python.lua
+++ b/lua/neogen/configurations/python.lua
@@ -101,6 +101,9 @@ return {
                        if res["return_type"] then
                            res["return_statement"] = nil
                            res["anonymous_return"] = nil
+                            if res["return_type"][1] == "None" then
+                                res["return_type"] = nil
+                            end
                        end

                        -- Remove reference to "self" if any