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)