Avoid crash when self.total_time_taken is None?

kundajelab · Jan 8, 2021 · e668030 · e668030
1 parent aeb0a20
commit e668030
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modisco/tfmodisco_workflow/seqlets_to_patterns.py b/modisco/tfmodisco_workflow/seqlets_to_patterns.py
@@ -594,8 +594,8 @@ def from_hdf5(cls, grp, track_set):
 
     def save_hdf5(self, grp):
         grp.attrs["success"] = self.success
-        grp.attrs["total_time_taken"] = self.total_time_taken
         if (self.success):
+            grp.attrs["total_time_taken"] = self.total_time_taken
             if (self.each_round_initcluster_motifs is not None):
                 self.save_each_round_initcluster_motifs(
                     grp=grp.create_group("each_round_initcluster_motifs"))