format

matsengrp · Nov 14, 2024 · 281e119 · 281e119
1 parent 749f038
commit 281e119
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/netam/dasm.py b/netam/dasm.py
@@ -117,6 +117,7 @@ def zap_predictions_along_diagonal(predictions, aa_parents_idxs):
 
 class DASMBurrito(framework.TwoLossMixin, DXSMBurrito):
     prefix = "dasm"
+
     def __init__(self, *args, loss_weights: list = [1.0, 0.01], **kwargs):
         super().__init__(*args, **kwargs)
         self.xent_loss = torch.nn.CrossEntropyLoss()

diff --git a/netam/dxsm.py b/netam/dxsm.py
@@ -32,6 +32,7 @@
 
 class DXSMDataset(Dataset, ABC):
     prefix = "dxsm"
+
     def __init__(
         self,
         nt_parents: pd.Series,
@@ -240,6 +241,7 @@ def update_neutral_probs(self):
 
 class DXSMBurrito(framework.Burrito, ABC):
     prefix = "dxsm"
+
     def _find_optimal_branch_length(
         self,
         parent,