pkl data added

Author: bartzbeielstein

pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "spotPython"
-version = "0.6.41"
+version = "0.6.42"
 authors = [
   { name="T. Bartz-Beielstein", email="tbb@bartzundbartz.de" }
 ]
@@ -64,9 +64,6 @@ include-package-data = true
 namespaces = true
 where = ["src"]
 
-[tool.setuptools.package-data]
-spotPython = ["*.json", "*.csv", "*.pkl"]
-
 [tool.black]
 line-length = 120
 target-version = ["py311"]

test/test_pkldataset.py

@@ -1,35 +1,34 @@
-# TODO: Setup test for PKLDataset that can use the pkl file form data folder
-# import pytest
-# import torch
-# from torch.utils.data import DataLoader
-# from spotPython.data.pkldataset import PKLDataset
+import pytest
+import torch
+from torch.utils.data import DataLoader
+from spotPython.data.pkldataset import PKLDataset
 
 
-# def test_pkl_dataset():
-#     # Create an instance of PKLDataset for testing
-#     dataset = PKLDataset(target_column='prognosis')
+def test_pkl_dataset():
+    # Create an instance of PKLDataset for testing
+    dataset = PKLDataset(target_column='prognosis')
 
-#     # Test the length of the dataset
-#     assert len(dataset) > 0
+    # Test the length of the dataset
+    assert len(dataset) > 0
 
-#     # Test __getitem__
-#     idx = 0
-#     sample = dataset[idx]
-#     assert isinstance(sample, tuple)
-#     assert len(sample) == 2
-#     feature, target = sample
-#     assert isinstance(feature, torch.Tensor)
-#     assert isinstance(target, torch.Tensor)
+    # Test __getitem__
+    idx = 0
+    sample = dataset[idx]
+    assert isinstance(sample, tuple)
+    assert len(sample) == 2
+    feature, target = sample
+    assert isinstance(feature, torch.Tensor)
+    assert isinstance(target, torch.Tensor)
 
-#     # Test DataLoader
-#     batch_size = 3
-#     dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=True)
-#     for batch in dataloader:
-#         inputs, targets = batch
-#         assert inputs.size(0) == batch_size
-#         assert targets.size(0) == batch_size
-#         break
+    # Test DataLoader
+    batch_size = 3
+    dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=True)
+    for batch in dataloader:
+        inputs, targets = batch
+        assert inputs.size(0) == batch_size
+        assert targets.size(0) == batch_size
+        break
 
 
-# if __name__ == "__main__":
-#     pytest.main(["-v", __file__])
+if __name__ == "__main__":
+    pytest.main(["-v", __file__])