python3 test.py
3.314018249511719e-05 Loading substs from  /home/y.kozhevnikov/Desktop/ezhik/dta_1-limitNone-maxexperwordNone/modelcheckpoints_alldata-small-xlmfinetuned_checkpoint36.pt/<mask><mask>-T-2ltr2f_topk150_fixspacesTrue.npz.input
Traceback (most recent call last):
  File "/home/y.kozhevnikov/anaconda3/lib/python3.7/site-packages/pandas/core/indexes/base.py", line 2646, in get_loc
    return self._engine.get_loc(key)
  File "pandas/_libs/index.pyx", line 111, in pandas._libs.index.IndexEngine.get_loc
  File "pandas/_libs/index.pyx", line 138, in pandas._libs.index.IndexEngine.get_loc
  File "pandas/_libs/hashtable_class_helper.pxi", line 1618, in pandas._libs.hashtable.PyObjectHashTable.get_item
  File "pandas/_libs/hashtable_class_helper.pxi", line 1626, in pandas._libs.hashtable.PyObjectHashTable.get_item
KeyError: '0'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "test.py", line 3, in <module>
    s = load_substs('/home/y.kozhevnikov/Desktop/ezhik/dta_1-limitNone-maxexperwordNone/modelcheckpoints_alldata-small-xlmfinetuned_checkpoint36.pt/<mask><mask>-T-2ltr2f_topk150_fixspacesTrue.npz.input')
  File "/home/y.kozhevnikov/Desktop/ezhik/substs_stat.py", line 16, in load_substs
    substs_probs = pd.read_csv(p, index_col=0, nrows=limit)['0']
  File "/home/y.kozhevnikov/anaconda3/lib/python3.7/site-packages/pandas/core/frame.py", line 2800, in __getitem__
    indexer = self.columns.get_loc(key)
  File "/home/y.kozhevnikov/anaconda3/lib/python3.7/site-packages/pandas/core/indexes/base.py", line 2648, in get_loc
    return self._engine.get_loc(self._maybe_cast_indexer(key))
  File "pandas/_libs/index.pyx", line 111, in pandas._libs.index.IndexEngine.get_loc
  File "pandas/_libs/index.pyx", line 138, in pandas._libs.index.IndexEngine.get_loc
  File "pandas/_libs/hashtable_class_helper.pxi", line 1618, in pandas._libs.hashtable.PyObjectHashTable.get_item
  File "pandas/_libs/hashtable_class_helper.pxi", line 1626, in pandas._libs.hashtable.PyObjectHashTable.get_item
KeyError: '0'