DW fails to display DataFrame in dict where key is Enum member

[danhje]

Environment data

• VS Code version: 1.93.1
• Data Wrangler Extension version (available under the Extensions sidebar): v1.8.1
• Jupyter Extension version (available under the Extensions sidebar): v2024.8.1
• Python Extension version (available under the Extensions sidebar): v2024.14.1
• OS (Windows | Mac | Linux distro) and version: Windows 11
• Pandas version: 2.2.2
• Python and/or Anaconda version: 3.12.5
• Type of virtual environment used (N/A | venv | virtualenv | conda | ...): poetry venv

Expected behaviour

DataFrame is displayed

Actual behaviour

Nothing happens

Steps to reproduce:

Debug this script with a breakpoint on pass:

from enum import Enum
import pandas as pd

class MyEnum(Enum):
    ONE = 1

df = pd.DataFrame({'a': [1, 2, 3], 'b': [4, 5, 6]})
d = {"ONE": df, MyEnum.ONE: df}

pass

The DF that corresponds to the "ONE" key opens without issue, but the DF that corresponds with the MyEnum.ONE value does not.

[pwang347]

Hi @danhje, thank you for reporting the issue! It looks like this one is also a problem with the Jupyter extension's variable delegation.

From Jupyter output logs (with logging level set to debug):

Working launch:

11:46:41.719 [debug] Evaluating in debugger : 60f5e28b-9a31-4f3b-b97d-e1c5c98594d1: _VSCODE_getVariable("info", True, d['ONE'])
11:46:41.771 [info] Showing data viewer with command dataWrangler.launchDataViewer for variable {"name":"d['ONE']","type":"DataFrame","count":3,"shape":"(3, 2)","size":0,"supportsDataExplorer":false,"value":"...","truncated":false,"frameId":8,"fullType":"pandas.core.frame.DataFrame"}

Failing launch (doesn't show launch message):

11:46:52.371 [debug] Evaluating in debugger : 60f5e28b-9a31-4f3b-b97d-e1c5c98594d1: _VSCODE_getVariable("info", True, <MyEnum.ONE: 1>)