Skip to content

Commit 8c18dd6

Browse files
test pkl data added to toml
1 parent f8968e1 commit 8c18dd6

2 files changed

Lines changed: 14 additions & 50 deletions

File tree

notebooks/00_spotPython_tests.ipynb

Lines changed: 8 additions & 48 deletions
Original file line numberDiff line numberDiff line change
@@ -137,19 +137,9 @@
137137
},
138138
{
139139
"cell_type": "code",
140-
"execution_count": 1,
140+
"execution_count": null,
141141
"metadata": {},
142-
"outputs": [
143-
{
144-
"name": "stdout",
145-
"output_type": "stream",
146-
"text": [
147-
"Loading data from /Users/bartz/miniforge3/envs/spotCondaEnv/lib/python3.11/site-packages/spotPython/data/data.csv\n",
148-
"torch.Size([11, 64])\n",
149-
"torch.Size([11])\n"
150-
]
151-
}
152-
],
142+
"outputs": [],
153143
"source": [
154144
"from spotPython.data.csvdataset import CSVDataset\n",
155145
"# dataset = CSVDataset(csv_file='./data/spotPython/data.csv', target_column='prognosis')\n",
@@ -160,48 +150,18 @@
160150
},
161151
{
162152
"cell_type": "code",
163-
"execution_count": 5,
153+
"execution_count": null,
164154
"metadata": {},
165-
"outputs": [
166-
{
167-
"data": {
168-
"text/plain": [
169-
"'Split: Train'"
170-
]
171-
},
172-
"execution_count": 5,
173-
"metadata": {},
174-
"output_type": "execute_result"
175-
}
176-
],
155+
"outputs": [],
177156
"source": [
178157
"dataset.extra_repr()"
179158
]
180159
},
181160
{
182161
"cell_type": "code",
183-
"execution_count": 6,
162+
"execution_count": null,
184163
"metadata": {},
185-
"outputs": [
186-
{
187-
"name": "stdout",
188-
"output_type": "stream",
189-
"text": [
190-
"Batch Size: 3\n",
191-
"---------------\n",
192-
"Inputs: tensor([[1, 1, 1, 1, 0, 1, 0, 1, 0, 1, 0, 0, 1, 1, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1,\n",
193-
" 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1,\n",
194-
" 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 1, 0, 0, 0, 0, 0],\n",
195-
" [0, 1, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 0, 0, 1, 1, 1, 0, 0,\n",
196-
" 1, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 1, 0, 0, 1, 0, 0, 0, 1, 1, 0, 0, 1,\n",
197-
" 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0],\n",
198-
" [1, 1, 1, 0, 0, 1, 1, 1, 0, 0, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1,\n",
199-
" 1, 0, 1, 1, 1, 0, 1, 0, 1, 0, 1, 1, 0, 1, 1, 0, 0, 0, 1, 0, 1, 1, 1, 0,\n",
200-
" 1, 1, 1, 1, 1, 1, 0, 1, 1, 0, 0, 1, 1, 0, 1, 1]])\n",
201-
"Targets: tensor([6, 8, 3])\n"
202-
]
203-
}
204-
],
164+
"outputs": [],
205165
"source": [
206166
"from torch.utils.data import DataLoader\n",
207167
"# Set batch size for DataLoader\n",
@@ -394,7 +354,7 @@
394354
},
395355
{
396356
"cell_type": "code",
397-
"execution_count": 4,
357+
"execution_count": 1,
398358
"metadata": {},
399359
"outputs": [],
400360
"source": [
@@ -405,7 +365,7 @@
405365
},
406366
{
407367
"cell_type": "code",
408-
"execution_count": 3,
368+
"execution_count": 2,
409369
"metadata": {},
410370
"outputs": [
411371
{

pyproject.toml

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
77

88
[project]
99
name = "spotPython"
10-
version = "0.6.39"
10+
version = "0.6.40"
1111
authors = [
1212
{ name="T. Bartz-Beielstein", email="tbb@bartzundbartz.de" }
1313
]
@@ -61,11 +61,15 @@ Repository = "https://github.com/sequential-parameter-optimization/spotPython"
6161
include-package-data = true
6262

6363
[tool.setuptools.packages.find]
64+
namespaces = true
6465
where = ["src"]
6566

67+
[tool.setuptools.package-data]
68+
spotPython = ["*.json", "*.csv", "*.pkl"]
69+
6670
[tool.black]
6771
line-length = 120
68-
target-version = ["py310"]
72+
target-version = ["py311"]
6973

7074
[tool.pytest.ini_options]
7175
addopts = [

0 commit comments

Comments
 (0)