diff --git a/VOID/dockers/mcdocker.py b/VOID/dockers/mcdocker.py index c817829..120de30 100644 --- a/VOID/dockers/mcdocker.py +++ b/VOID/dockers/mcdocker.py @@ -32,3 +32,11 @@ def dock(self, attempts): def copy(self): return self.__class__(**self.__dict__.copy()) + + def on_start(self, cpx): + v = np.random.rand(1, 3) + lattice = cpx.host.lattice.matrix + translation = (v @ lattice).reshape(-1) + + cpx.guest_transform.translate(translation) + return cpx diff --git a/VOID/dockers/tests/test_mcdocker.py b/VOID/dockers/tests/test_mcdocker.py index 3499c9a..f621f71 100644 --- a/VOID/dockers/tests/test_mcdocker.py +++ b/VOID/dockers/tests/test_mcdocker.py @@ -17,7 +17,7 @@ def setUp(self): self.fitness = MinDistanceFitness(threshold=1.5) self.num_steps = 100 - self.temperature = 0 + self.temperature = 0.1 self.mcdocker = MonteCarloDocker( self.host, self.guest,