Skip to content

store_multi_timeseries_df fails with get_multi_timeseries_df output #261

@jbkolze

Description

@jbkolze

Running the following code (with ts_update_list being a list of time series id strings):

cwmspy_activate_t7()
t7_data = cwms.get_multi_timeseries_df(ts_update_list, "LRL")

cwmspy_activate_test(CDA_API_KEY_TEST)
cwms.store_multi_timeseries_df(ts_update_list, "LRL")

results in this error:

Traceback (most recent call last):
  File "/home/bkolze/projects/cwbi-utilities/.venv/lib/python3.9/site-packages/pandas/core/indexes/base.py", line 3805, in get_loc
    return self._engine.get_loc(casted_key)
  File "index.pyx", line 167, in pandas._libs.index.IndexEngine.get_loc
  File "index.pyx", line 196, in pandas._libs.index.IndexEngine.get_loc
  File "pandas/_libs/hashtable_class_helper.pxi", line 7081, in pandas._libs.hashtable.PyObjectHashTable.get_item
  File "pandas/_libs/hashtable_class_helper.pxi", line 7089, in pandas._libs.hashtable.PyObjectHashTable.get_item
KeyError: 'ts_id'

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

Traceback (most recent call last):
  File "/home/bkolze/projects/cwbi-utilities/t7_copy_dlb/copy_dlb_cwbi_test.py", line 130, in <module>
    copy_dlb_t7_to_test()
  File "/home/bkolze/projects/cwbi-utilities/t7_copy_dlb/copy_dlb_cwbi_test.py", line 84, in copy_dlb_t7_to_test
    cwms.store_multi_timeseries_df(t7_data, "LRL")
  File "/home/bkolze/projects/cwbi-utilities/.venv/lib/python3.9/site-packages/cwms/timeseries/timeseries.py", line 483, in store_multi_timeseries_df
    ts_data_all["ts_id"].astype(str) + ":" + ts_data_all["version_date"].astype(str)
  File "/home/bkolze/projects/cwbi-utilities/.venv/lib/python3.9/site-packages/pandas/core/frame.py", line 4101, in __getitem__
    return self._getitem_multilevel(key)
  File "/home/bkolze/projects/cwbi-utilities/.venv/lib/python3.9/site-packages/pandas/core/frame.py", line 4159, in _getitem_multilevel
    loc = self.columns.get_loc(key)
  File "/home/bkolze/projects/cwbi-utilities/.venv/lib/python3.9/site-packages/pandas/core/indexes/multi.py", line 3040, in get_loc
    loc = self._get_level_indexer(key, level=0)
  File "/home/bkolze/projects/cwbi-utilities/.venv/lib/python3.9/site-packages/pandas/core/indexes/multi.py", line 3391, in _get_level_indexer
    idx = self._get_loc_single_level_index(level_index, key)
  File "/home/bkolze/projects/cwbi-utilities/.venv/lib/python3.9/site-packages/pandas/core/indexes/multi.py", line 2980, in _get_loc_single_level_index
    return level_index.get_loc(key)
  File "/home/bkolze/projects/cwbi-utilities/.venv/lib/python3.9/site-packages/pandas/core/indexes/base.py", line 3812, in get_loc
    raise KeyError(key) from err
KeyError: 'ts_id'

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions