File tree Expand file tree Collapse file tree 1 file changed +9
-5
lines changed
torchstain/base/normalizers Expand file tree Collapse file tree 1 file changed +9
-5
lines changed Original file line number Diff line number Diff line change 1- def MultiMacenkoNormalizer (backend = 'torch' , ** kwargs ):
2- if backend == 'torch' :
3- from torchstain .torch .normalizers .multitarget import MultiMacenkoNormalizer
4- return MultiMacenkoNormalizer (** kwargs )
1+ def MultiMacenkoNormalizer (backend = "torch" , ** kwargs ):
2+ if backend == "numpy" :
3+ raise NotImplementedError ("MultiMacenkoNormalizer is not implemented for NumPy backend" )
4+ elif backend == "torch" :
5+ from torchstain .torch .normalizers import TorchMultiMacenkoNormalizer
6+ return TorchMultiMacenkoNormalizer (** kwargs )
7+ elif backend == "tensorflow" :
8+ raise NotImplementedError ("MultiMacenkoNormalizer is not implemented for TensorFlow backend" )
59 else :
6- raise Exception (f' Unsupported backend { backend } ' )
10+ raise Exception (f" Unsupported backend { backend } " )
You can’t perform that action at this time.
0 commit comments