diff --git a/tracking/btree.py b/tracking/btree.py index 3325325..9ad27f5 100644 --- a/tracking/btree.py +++ b/tracking/btree.py @@ -123,6 +123,7 @@ class TrackingStorage(BTreeContainer): runs = None # currently active runs finishedRuns = None # finished runs currentRuns = None # the currently active run for each task + storage = None # compatibility with new cco.storage def __init__(self, *args, **kw): trackFactory = kw.pop('trackFactory', None) @@ -221,6 +222,9 @@ class TrackingStorage(BTreeContainer): self.indexTrack(trackNum, track) return trackId + def setTrackData(self, track, data): + track.data = data # persistent track: data will be stored automatically + def updateTrack(self, track, data, overwrite=False): trackId = str(track.__name__) trackNum = int(trackId)