Commit
·
5cc4a9e
1
Parent(s):
3a64edc
fix: Resolve NameError in get_dataset MCP tool
Browse filesFixed bug where comment and code were on same line causing dataset variable to never be assigned. This was causing 'name dataset is not defined' error when calling get_dataset tool.
- mcp_tools.py +2 -1
mcp_tools.py
CHANGED
|
@@ -1057,7 +1057,8 @@ async def get_dataset(
|
|
| 1057 |
"data": []
|
| 1058 |
}, indent=2)
|
| 1059 |
|
| 1060 |
-
# Load dataset from HuggingFace
|
|
|
|
| 1061 |
df = pd.DataFrame(dataset)
|
| 1062 |
|
| 1063 |
if df.empty:
|
|
|
|
| 1057 |
"data": []
|
| 1058 |
}, indent=2)
|
| 1059 |
|
| 1060 |
+
# Load dataset from HuggingFace
|
| 1061 |
+
dataset = load_dataset(dataset_repo, split="train")
|
| 1062 |
df = pd.DataFrame(dataset)
|
| 1063 |
|
| 1064 |
if df.empty:
|