diff --git a/taufactor/taufactor.py b/taufactor/taufactor.py index ab0e77e..a11322f 100644 --- a/taufactor/taufactor.py +++ b/taufactor/taufactor.py @@ -397,7 +397,6 @@ def __init__(self, img, cond={1: 1}, bc=(-0.5, 0.5), device=torch.device('cuda:0 for a 2 phase material, {1:0.543, 2: 0.420}, with 1s and 2s in the input img :param bc: Upper and lower boundary conditions. Leave as default. """ - super().__init__(img, bc, device) if (0 in cond.values()): raise ValueError( '0 conductivity phase: non-conductive phase should be labelled 0 in the input image and ommitted from the cond argument') @@ -408,6 +407,7 @@ def __init__(self, img, cond={1: 1}, bc=(-0.5, 0.5), device=torch.device('cuda:0 self.cond = {ph: 0.5 / c for ph, c in cond.items()} # Results + super().__init__(img, bc, device) self.pre_factors = self.nn[1:] self.nn = self.nn[0]