File tree Expand file tree Collapse file tree 2 files changed +9
-2
lines changed
Expand file tree Collapse file tree 2 files changed +9
-2
lines changed Original file line number Diff line number Diff line change 3838from aidsorb .utils import pcd_from_dir
3939
4040
41+ def dummy_tfm (x ):
42+ return x - 1
43+
4144class 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
Original file line number Diff line number Diff line change 3131from aidsorb .transforms import Center , RandomRotation
3232
3333
34+ def dummy_tfm (x ):
35+ return x - 1
36+
37+
3438class 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
You can’t perform that action at this time.
0 commit comments