one small change

matsengrp · Nov 22, 2024 · 8ca6e52 · 8ca6e52
1 parent f66a530
commit 8ca6e52
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/netam/dasm.py b/netam/dasm.py
@@ -11,6 +11,7 @@
 import netam.framework as framework
 import netam.molevol as molevol
 import netam.sequences as sequences
+import copy
 
 
 class DASMDataset(DXSMDataset):

diff --git a/netam/dxsm.py b/netam/dxsm.py
@@ -4,7 +4,6 @@
 from functools import partial
 
 import torch
-from torch.utils.data import Dataset
 
 # Amazingly, using one thread makes things 50x faster for branch length
 # optimization on our server.