Error detected while processing InsertEnter Autocommands: E117: Unknown function: jedi#remove_usages - vim

Question

I'm trying to setup vim-jedi on neovim.
Whenever I open a python file and switch to Insert mode I get the following error:
Error detected while processing InsertEnter Autocommands for "<buffer=1>":
E117: Unknown function: jedi#remove_usages
I'm really new to (neo)vim so I'd really appreciate if could help me resolve this issue, thanks!
Below, I'm providing the information I think is relevant. Please let me know if you require me to provide anything else.
My setup
First I installed vim-jedi from pacman then I set my init.vim file to the following.
" My .nvimrc(init.vim) file
"
:set number relativenumber
:set termguicolors
" Plugins
call plug#begin()
" Solarized theme
let g:neosolarized_contrast = "normal"
set background=dark
Plug 'iCyMind/NeoSolarized'
" Python Completions via jedi
Plug 'davidhalter/jedi-vim', { 'for': 'python' }
call plug#end()
" Select Solarized as my theme.
colorscheme NeoSolarized
Afterwards I
Ran :PlugInstall
Restarted nvim
Ran :PlugUpdate
Restarted nvim and opened my file with nvim db.py.
The file I opened contains the following:
"""Database management functions"""
import tasks3.tasks3.db as db
def init(db_path: str):
engine = db.create_engine(db_path)
Base.metadata.create_all(bind=engine)
def purge(db_path: str):
engine = db.create_engine(db_path)
session = db.orm.sessionmaker(bind=engine)
def drop(db_path: str):
pass
To the best of my knowledge I don't have any additional scripts for nvim.
Additional Note (not sure if this is related)
In the context of the above purge function if type session. I get the following message:
-- Omni completion (^O^N^P) Pattern not found