"...llm/tio/git@developer.sourcefind.cn:OpenDAS/dynamo.git" did not exist on "32a748e4030859dbb2a4dd9eaac389e2c84966b3"
Commit a317ad27 authored by Jennifer's avatar Jennifer
Browse files

superimposition fix from Aymen

parent 36cd9eb0
...@@ -35,10 +35,10 @@ def _superimpose_np(reference, coords): ...@@ -35,10 +35,10 @@ def _superimpose_np(reference, coords):
def _superimpose_single(reference, coords): def _superimpose_single(reference, coords):
reference_np = reference.detach().cpu().numpy() reference_np = reference.detach().to(torch.float).cpu().numpy()
coords_np = coords.detach().cpu().numpy() coords_np = coords.detach().to(torch.float).cpu().numpy()
superimposed, rmsd = _superimpose_np(reference_np, coords_np) superimposed, rmsd = _superimpose_np(reference_np, coords_np)
return coords.new_tensor(superimposed), coords.new_tensor(rmsd) return coords.new_tensor(superimposed), coords.new_tensor(rmsd)
def superimpose(reference, coords, mask): def superimpose(reference, coords, mask):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment