UMAP points throws index out of bounds on sentiment classification dataset

[mikeldking]

Sentiment classification dataset throws errors on the embeddings view: http://localhost:6060/embeddings/RW1iZWRkaW5nRGltZW5zaW9uOjA=

index 31410 is out of bounds for axis 0 with size 0

GraphQL request:8:7
7 |     ... on EmbeddingDimension {
8 |       UMAPPoints(timeRange: $timeRange) {
  |       ^
9 |         data {
Traceback (most recent call last):
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/graphql/execution/execute.py", line 521, in execute_field
    result = resolve_fn(source, info, **args)
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/strawberry/schema/schema_converter.py", line 519, in _resolver
    return _get_result(_source, strawberry_info, **kwargs)
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/strawberry/schema/schema_converter.py", line 511, in _get_result
    return field.get_result(
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/strawberry/field.py", line 169, in get_result
    return self.base_resolver(*args, **kwargs)
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/strawberry/types/fields/resolver.py", line 186, in __call__
    return self.wrapped_func(*args, **kwargs)
  File "/Users/mikeldking/work/phoenix/src/phoenix/server/api/types/EmbeddingDimension.py", line 321, in UMAPPoints
    link_to_data = link_to_data_column[row_id]
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/pandas/core/series.py", line 978, in __getitem__
    return self._values[key]
IndexError: index 31410 is out of bounds for axis 0 with size 0
Stack (most recent call last):
  File "/Users/mikeldking/.pyenv/versions/3.10.3/lib/python3.10/runpy.py", line 196, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File "/Users/mikeldking/.pyenv/versions/3.10.3/lib/python3.10/runpy.py", line 86, in _run_code
    exec(code, run_globals)
  File "/Users/mikeldking/work/phoenix/src/phoenix/server/main.py", line 81, in <module>
    uvicorn.run(app, port=args.port)
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/uvicorn/main.py", line 569, in run
    server.run()
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/uvicorn/server.py", line 60, in run
    return asyncio.run(self.serve(sockets=sockets))
  File "/Users/mikeldking/.pyenv/versions/3.10.3/lib/python3.10/asyncio/runners.py", line 44, in run
    return loop.run_until_complete(main)
  File "/Users/mikeldking/.pyenv/versions/3.10.3/lib/python3.10/asyncio/base_events.py", line 633, in run_until_complete
    self.run_forever()
  File "/Users/mikeldking/.pyenv/versions/3.10.3/lib/python3.10/asyncio/base_events.py", line 600, in run_forever
    self._run_once()
  File "/Users/mikeldking/.pyenv/versions/3.10.3/lib/python3.10/asyncio/base_events.py", line 1896, in _run_once
    handle._run()
  File "/Users/mikeldking/.pyenv/versions/3.10.3/lib/python3.10/asyncio/events.py", line 80, in _run
    self._context.run(self._callback, *self._args)
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/starlette/middleware/base.py", line 69, in coro
    await self.app(scope, receive_or_disconnect, send_no_error)
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/starlette/middleware/exceptions.py", line 68, in __call__
    await self.app(scope, receive, sender)
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/starlette/routing.py", line 706, in __call__
    await route.handle(scope, receive, send)
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/starlette/routing.py", line 276, in handle
    await self.app(scope, receive, send)
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/strawberry/asgi/__init__.py", line 48, in __call__
    await self.http_handler_class(
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/strawberry/asgi/handlers/http_handler.py", line 50, in handle
    response = await self.get_http_response(
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/strawberry/asgi/handlers/http_handler.py", line 148, in get_http_response
    result = await execute(
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/strawberry/asgi/handlers/http_handler.py", line 201, in execute
    return await self.schema.execute(
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/strawberry/schema/schema.py", line 226, in execute
    result = await execute(
  File "/Users/mikeldking/.virtualenvs/phoenix/lib/python3.10/site-packages/strawberry/schema/execute.py", line 144, in execute
    process_errors(result.errors, execution_context)
    ```