1 files changed, 8 insertions, 4 deletions
diff --git a/lua/lsp/python-ls.lua b/lua/lsp/python-ls.lua
index 4898a0b5..4bdbd9d9 100644
--- a/lua/lsp/python-ls.lua
+++ b/lua/lsp/python-ls.lua
@@ -1,8 +1,12 @@
 -- npm i -g pyright
--- local capabilities = vim.lsp.protocol.make_client_capabilities()
--- capabilities.textDocument.completion.completionItem.snippetSupport = true
 require'lspconfig'.pyright.setup {
     cmd = {DATA_PATH .. "/lspinstall/python/node_modules/.bin/pyright-langserver", "--stdio"},
-    on_attach = require'lsp'.common_on_attach
-    -- capabilities = capabilities
+    on_attach = require'lsp'.common_on_attach,
+     handlers = {
+       ["textDocument/publishDiagnostics"] = vim.lsp.with(
+         vim.lsp.diagnostic.on_publish_diagnostics, {
+           virtual_text = O.python.virtual_text
+         }
+       ),
+     }
 }