kshitijthakkar commited on
Commit
5cc4a9e
·
1 Parent(s): 3a64edc

fix: Resolve NameError in get_dataset MCP tool

Browse files

Fixed 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.

Files changed (1) hide show
  1. 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 dataset = load_dataset(dataset_repo, split="train")
 
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: