Skip to content

Commit f268a78

Browse files
committed
Update base multitarget
1 parent 7b2620e commit f268a78

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed
Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,10 @@
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}")

0 commit comments

Comments
 (0)