Skip to content

Commit dc368e7

Browse files
committed
refactor(tests): replace lambdas with top-level functions
This change ensures that unit tests pass in Python 3.14, where the forkserver start method is used by default and requires picklable objects.
1 parent e691932 commit dc368e7

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

tests/test_data.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,9 @@
3838
from aidsorb.utils import pcd_from_dir
3939

4040

41+
def dummy_tfm(x):
42+
return x - 1
43+
4144
class TestPrepareData(unittest.TestCase):
4245
def setUp(self):
4346
self.tempdir = tempfile.TemporaryDirectory(dir='/tmp')
@@ -82,7 +85,7 @@ def setUp(self):
8285
self.index_col = 'id'
8386
self.labels = ['y1', 'y3']
8487
self.transform_x = Center()
85-
self.transform_y = lambda y: y - 1
88+
self.transform_y = dummy_tfm
8689
self.batch_size = 2
8790
self.channels_first = True
8891

tests/test_datamodules.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,13 +31,17 @@
3131
from aidsorb.transforms import Center, RandomRotation
3232

3333

34+
def dummy_tfm(x):
35+
return x - 1
36+
37+
3438
class TestPCDDataModule(unittest.TestCase):
3539
def setUp(self):
3640
# Arguments for the datamodule.
3741
self.train_size = 2
3842
self.train_trans_x = Center()
3943
self.eval_trans_x = RandomRotation()
40-
self.trans_y = lambda y: y -1
44+
self.trans_y = dummy_tfm
4145
self.shuffle = True
4246
self.drop_last = True
4347
self.train_bs = 3

0 commit comments

Comments
 (0)