def test_execute_query_custom_status(): data = [ Trial(name=str(i), status=CustomStatus('new', 1)) for i in range(0, 20) ] + [ Trial(name=str(i), status=CustomStatus('interrupted', 1)) for i in range(0, 20) ] query = dict(status={'$in': ['new', 'interrupted']}) arr = check_query(data, query) assert len(arr) == 40
def make_status(status): if status is None: return None if status['name'] in _STATUS_STR: return Status(status['value']) return CustomStatus(name=status['name'], value=status['value'])
import os from track.persistence import get_protocol from track.structure import Trial, TrialGroup, Project, Status, CustomStatus new = CustomStatus('new', Status.CreatedGroup.value + 1) reserved = CustomStatus('reserved', Status.CreatedGroup.value + 2) project = Project(name='test') group = TrialGroup(name='MyGroup', project_id=project.uid) statuses = [ Status.Interrupted, Status.Broken, Status.Completed, reserved, new, new, new, new, new ] trials = [] for idx, status in enumerate(statuses): trial = Trial(project_id=project.uid, group_id=group.uid, status=Status.Interrupted, parameters={ 'batch_size': 256, 'id': idx }) trials.append(trial) TRIAL_COUNT = len(trials) def remove(filename):
HAS_TRACK = True except ImportError: REASON = "Track is not installed" except SyntaxError: major, minor, patch, _, _ = sys.version_info if minor < 6: REASON = "Python is too old" log.warning("Track does not support python < 3.6!") else: raise if HAS_TRACK: _status = [ CustomStatus("new", TrackStatus.CreatedGroup.value + 1), CustomStatus("reserved", TrackStatus.CreatedGroup.value + 2), ] _status_dict = {s.name: s for s in _status} _status_dict["completed"] = TrackStatus.Completed _status_dict["interrupted"] = TrackStatus.Interrupted _status_dict["broken"] = TrackStatus.Broken _status_dict["suspended"] = TrackStatus.Suspended def get_track_status(val): """Convert orion status to track status""" return _status_dict.get(val)
HAS_TRACK = True except ImportError: REASON = 'Track is not installed' except SyntaxError: major, minor, patch, _, _ = sys.version_info if minor < 6: REASON = 'Python is too old' log.warning('Track does not support python < 3.6!') else: raise if HAS_TRACK: _status = [ CustomStatus('new', TrackStatus.CreatedGroup.value + 1), CustomStatus('reserved', TrackStatus.CreatedGroup.value + 2), ] _status_dict = {s.name: s for s in _status} _status_dict['completed'] = TrackStatus.Completed _status_dict['interrupted'] = TrackStatus.Interrupted _status_dict['broken'] = TrackStatus.Broken _status_dict['suspended'] = TrackStatus.Suspended def get_track_status(val): """Convert orion status to track status""" return _status_dict.get(val)