Hi folks. I have the following problem.
I am getting this error while I run it with pyflyte run --remote train.py training_workflow command.
When I run the script locally (on my machine without pyflyte) everything works perfectly.
I'd appreciate any help.
Cheers.
Traceback (most recent call last):
File "/usr/local/lib/python3.10/site-packages/flytekit/exceptions/scopes.py", line 206, in user_entry_point
return wrapped(*args, **kwargs)
File "/root/flyteWorkFlow/train.py", line 69, in get_data
all_docs['class_int'] = pd.Categorical(all_docs['class']).codes
File "/usr/local/lib/python3.10/site-packages/pandas/core/frame.py", line 3807, in getitem
indexer = self.columns.get_loc(key)
File "/usr/local/lib/python3.10/site-packages/pandas/core/indexes/base.py", line 3804, in get_loc
raise KeyError(key) from err
Message:
'class'
User error.