.

mims-harvard · Feb 16, 2025 · 2cdb0bc · 2cdb0bc
1 parent 261adcd
commit 2cdb0bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tdc/test/test_model_server.py b/tdc/test/test_model_server.py
@@ -84,7 +84,7 @@ def testGeneformerPerturb(self):
         assert input_tensor.shape[0] == 512, "unexpected batch size"
         assert input_tensor.shape[1] == mdim, f"unexpected gene length {mdim}"
         attention_mask = torch.tensor([[t != 0 for t in cell] for cell in batch
-                                       ])
+                                      ])
         assert input_tensor.shape[0] == attention_mask.shape[0]
         assert input_tensor.shape[1] == attention_mask.shape[1]
         try: