leigh/neogen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ref: Use typed parameters

Browse Source
This commit is contained in:
danymat
2023-01-12 22:30:28 +01:00
parent 6514bb1b50
commit ffefbd27ba
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lua/neogen/configurations/python.lua
View File

@@ -127,7 +127,7 @@ return {
-- Check if the function is inside a class
-- If so, remove reference to the first parameter (that can be `self`, `cls`, or a custom name)
if res.parameters and locator({ current = node }, parent.class) then
if res[i.Parameter] and locator({ current = node }, parent.class) then
local remove_identifier = true
-- Check if function is a static method. If so, will not remove the first parameter
if node:parent():type() == "decorated_definition" then
@@ -138,9 +138,9 @@ return {
end
end
if remove_identifier then
table.remove(res.parameters, 1)
if vim.tbl_isempty(res.parameters) then
res.parameters = nil
table.remove(res[i.Parameter], 1)
if vim.tbl_isempty(res[i.Parameter]) then
res[i.Parameter] = nil
end
end
end