Skip to content

Commit f6becc9

Browse files
authored
Merge pull request #316 from pimlock/pmlocek/tool-calling-python-sdk
feat: update data-flywheel/tool-calling notebooks to use Python SDK
2 parents 54d27c6 + d0f67e7 commit f6becc9

File tree

7 files changed

+927
-1178
lines changed

7 files changed

+927
-1178
lines changed

nemo/data-flywheel/tool-calling/1_data_preparation.ipynb

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,6 @@
3838
"from typing import Any, Dict, List, Union\n",
3939
"\n",
4040
"import numpy as np\n",
41-
"import torch\n",
4241
"from datasets import load_dataset"
4342
]
4443
},
@@ -62,8 +61,6 @@
6261
"# Limits to at most N tool properties\n",
6362
"LIMIT_TOOL_PROPERTIES = 8\n",
6463
"\n",
65-
"torch.manual_seed(SEED)\n",
66-
"torch.cuda.manual_seed_all(SEED)\n",
6764
"np.random.seed(SEED)\n",
6865
"random.seed(SEED)"
6966
]
@@ -675,7 +672,7 @@
675672
"name": "python",
676673
"nbconvert_exporter": "python",
677674
"pygments_lexer": "ipython3",
678-
"version": "3.12.3"
675+
"version": "3.11.13"
679676
}
680677
},
681678
"nbformat": 4,

0 commit comments

Comments
 (0)