From 0d381d03d2bb3b3e661b79a8e16549ce9484e13d Mon Sep 17 00:00:00 2001
From: danymat <d.danymat@gmail.com>
Date: Sat, 29 Jan 2022 19:56:14 +0100
Subject: [PATCH] feat(py) `None` return type does not display returns (#51)

---
 lua/neogen/configurations/python.lua | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/lua/neogen/configurations/python.lua b/lua/neogen/configurations/python.lua
index 8849900..9aad0b3 100644
--- 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