search

cellarium-ai / cellarium-ml

Distributed single-cell data analysis.
BSD 3-Clause "New" or "Revised" License
22 stars 3 forks source link

Add a teardown method for `cpu_transforms` to handle multi-use case of a `CellariumModule` object #244

Closed ordabayevy closed 1 month ago

ordabayevy commented 1 month ago

For example if you train a CellariumModule with cpu_transforms with one datamodule and then use another datamodule for prediction then cpu_transforms will stay attached to the training datamodule. The detaching mechanism needs to be added at the end of training/predicting.