def test_resolve_filespecs_raises_with_bad_inputs(self, task_id, outputs):
with self.assertRaises(ValueError):
io.resolve_filespecs(task_id, *outputs)
def default_options(add_flags=True, flags_obj=None):
"""Creates a DeepVariantOptions proto populated with reasonable defaults.
Args:
add_flags: bool. defaults to True. If True, we will push the value of
certain FLAGS into our options. If False, those option fields are left
uninitialized.
flags_obj: object. If not None, use as the source of flags,
else use global FLAGS.
Returns:
deepvariant_pb2.DeepVariantOptions protobuf.
Raises:
ValueError: If we observe invalid flag values.
"""
if not flags_obj:
flags_obj = FLAGS
read_reqs = reads_pb2.ReadRequirements(
min_base_quality=10,
min_mapping_quality=10,
min_base_quality_mode=reads_pb2.ReadRequirements.ENFORCED_BY_CLIENT)
pic_options = pileup_image.default_options(read_requirements=read_reqs)
allele_counter_options = deepvariant_pb2.AlleleCounterOptions(
partition_size=flags_obj.partition_size, read_requirements=read_reqs)
if flags_obj.sample_name:
sample_name = flags_obj.sample_name
elif flags_obj.reads:
with sam.SamReader(flags_obj.reads) as sam_reader:
sample_name = extract_sample_name_from_sam_reader(sam_reader)
else:
sample_name = _UNKNOWN_SAMPLE
variant_caller_options = deepvariant_pb2.VariantCallerOptions(
min_count_snps=flags_obj.vsc_min_count_snps,
min_count_indels=flags_obj.vsc_min_count_indels,
min_fraction_snps=flags_obj.vsc_min_fraction_snps,
min_fraction_indels=flags_obj.vsc_min_fraction_indels,
# Not specified by default: fraction_reference_sites_to_emit,
# Fixed random seed produced with 'od -vAn -N4 -tu4 < /dev/urandom'.
random_seed=1400605801,
sample_name=sample_name,
p_error=0.001,
max_gq=50,
gq_resolution=flags_obj.gvcf_gq_binsize,
ploidy=2)
options = deepvariant_pb2.DeepVariantOptions(
exclude_contigs=exclude_contigs.EXCLUDED_HUMAN_CONTIGS,
# Fixed random seed produced with 'od -vAn -N4 -tu4 < /dev/urandom'.
random_seed=609314161,
# # Not specified by default: calling_regions = 3;
read_requirements=read_reqs,
allele_counter_options=allele_counter_options,
variant_caller_options=variant_caller_options,
pic_options=pic_options,
n_cores=1,
task_id=0,
num_shards=0,
min_shared_contigs_basepairs=0.9,
)
if add_flags:
options.mode = parse_proto_enum_flag(
deepvariant_pb2.DeepVariantOptions.Mode, flags_obj.mode.upper())
options.labeler_algorithm = parse_proto_enum_flag(
deepvariant_pb2.DeepVariantOptions.LabelerAlgorithm,
flags_obj.labeler_algorithm.upper())
if flags_obj.ref:
options.reference_filename = flags_obj.ref
if flags_obj.reads:
options.reads_filename = flags_obj.reads
if flags_obj.confident_regions:
options.confident_regions_filename = flags_obj.confident_regions
if flags_obj.truth_variants:
options.truth_variants_filename = flags_obj.truth_variants
if flags_obj.downsample_fraction != NO_DOWNSAMPLING:
options.downsample_fraction = flags_obj.downsample_fraction
if flags_obj.multi_allelic_mode:
multi_allelic_enum = {
'include_het_alt_images':
deepvariant_pb2.PileupImageOptions.ADD_HET_ALT_IMAGES,
'exclude_het_alt_images':
deepvariant_pb2.PileupImageOptions.NO_HET_ALT_IMAGES,
}[flags_obj.multi_allelic_mode]
options.pic_options.multi_allelic_mode = multi_allelic_enum
if flags_obj.pileup_image_height:
options.pic_options.height = flags_obj.pileup_image_height
if flags_obj.pileup_image_width:
options.pic_options.width = flags_obj.pileup_image_width
num_shards, examples, candidates, gvcf = io_utils.resolve_filespecs(
flags_obj.task, flags_obj.examples or '', flags_obj.candidates or '',
flags_obj.gvcf or '')
options.examples_filename = examples
options.candidates_filename = candidates
options.gvcf_filename = gvcf
options.calling_regions.extend(parse_regions_flag(flags_obj.regions))
options.exclude_calling_regions.extend(
parse_regions_flag(flags_obj.exclude_regions))
options.task_id = flags_obj.task
options.num_shards = 0 if num_shards is None else num_shards
options.realigner_enabled = flags_obj.realign_reads
if options.realigner_enabled:
options.realigner_options.CopyFrom(realigner.realigner_config(flags_obj))
options.max_reads_per_partition = flags_obj.max_reads_per_partition
if (options.mode == deepvariant_pb2.DeepVariantOptions.TRAINING and
flags_obj.training_random_emit_ref_sites != NO_RANDOM_REF):
options.variant_caller_options.fraction_reference_sites_to_emit = (
flags_obj.training_random_emit_ref_sites)
return options
def test_resolve_filespecs(self, task_id, outputs, expected): self.assertEqual(io.resolve_filespecs(task_id, *outputs), expected)