KeyError

[madkuchinmayarao]

2023-05-06 10:39:50.429221: I tensorflow/core/platform/cpu_feature_guard.cc:182] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
2023-05-06 10:39:51.623062: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT
Saved KNN Classifier not found....
Downloading MNIST Data, training KNN classifier and saving as knn.sav......
Kindly wait for a few minutes............
/usr/local/lib/python3.10/dist-packages/sklearn/datasets/_openml.py:968: FutureWarning: The default value of parser will change from 'liac-arff' to 'auto' in 1.4. You can set parser='auto' to silence this warning. Therefore, an ImportError will be raised from 1.4 if the dataset is dense and pandas is not installed. Note that the pandas parser may return different data types. See the Notes Section in fetch_openml's API doc for details.
warn(
Traceback (most recent call last):
File "/usr/local/lib/python3.10/dist-packages/pandas/core/indexes/base.py", line 3802, in get_loc
return self._engine.get_loc(casted_key)
File "pandas/_libs/index.pyx", line 138, in pandas._libs.index.IndexEngine.get_loc
File "pandas/_libs/index.pyx", line 165, in pandas._libs.index.IndexEngine.get_loc
File "pandas/_libs/hashtable_class_helper.pxi", line 5745, in pandas._libs.hashtable.PyObjectHashTable.get_item
File "pandas/_libs/hashtable_class_helper.pxi", line 5753, in pandas._libs.hashtable.PyObjectHashTable.get_item
KeyError: 34540

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "/content/AI_Sudoku/Run.py", line 19, in
knnobj.skl_knn()
File "/content/AI_Sudoku/KNN.py", line 40, in skl_knn
fifty_x, fifty_y = self.mk_dataset(50000)
File "/content/AI_Sudoku/KNN.py", line 28, in mk_dataset
train_img = [self.data[i] for i in self.indx[:size]]
File "/content/AI_Sudoku/KNN.py", line 28, in
train_img = [self.data[i] for i in self.indx[:size]]
File "/usr/local/lib/python3.10/dist-packages/pandas/core/frame.py", line 3807, in getitem
indexer = self.columns.get_loc(key)
File "/usr/local/lib/python3.10/dist-packages/pandas/core/indexes/base.py", line 3804, in get_loc
raise KeyError(key) from err
KeyError: 34540