Update dcph_utilities.py

autonlab · Apr 26, 2023 · ee77a7b · ee77a7b
1 parent 2c5a5ce
commit ee77a7b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/auton_survival/models/cph/dcph_utilities.py b/auton_survival/models/cph/dcph_utilities.py
@@ -10,6 +10,8 @@
 
 from auton_survival.models.dsm.utilities import get_optimizer, _reshape_tensor_with_nans
 
+from copy import deepcopy
+
 def randargmax(b,**kw):
   """ a random tie-breaking argmax"""
   return np.argmax(np.random.random(b.shape) * (b==b.max()), **kw)