Is there non-verbose mode to sleap loading a file?

[lisadiez]

Hello
Is there a way to turn off the metadata printing for the sleap_load command? We have thousands of files and would like to track progress/bugs etc, and all this extra text is making it tricky to follow. Any advice?
Thanks so much!

slp = sleap.load_file(fname_slp)

... OUTPUTS

video_path: /vast/cm5635/cohort2/huddles/2020_07_22_15_52_33_369348_compressed_defished_shrink_cropped.mp4
video_ind: 0
provenance: {}
Saved as /mnt/b3a68699-495d-4ebb-9ab1-ac74f11c68c5/gerbil/cohort2/huddles/2020_07_22_15_52_33_369348_compressed_Day_huddle.h5
track_occupancy: (1, 28802)
point_scores: (28802, 1, 1)
instance_scores: (28802, 1)
tracking_scores: (28802, 1)
labels_path: None

[roomrys]

Hi @lisadiez,

This looks like the output from writing an analysis file:

sleap/sleap/info/write_tracking_h5.py

Lines 242 to 286 in 57ee08a

	def write_occupancy_file(
	output_path: str, data_dict: Dict[str, Any], transpose: bool = True
	):
	"""Write HDF5 file with data from given dictionary.
	Args:
	output_path: Path of HDF5 file.
	data_dict: Dictionary with data to save. Keys are dataset names,
	values are the data.
	transpose: If True, then any ndarray in data dictionary will be
	transposed before saving. This is useful for writing files
	that will be imported into MATLAB, which expects data in
	column-major format.
	Returns:
	None
	"""
	with h5.File(output_path, "w") as f:
	print(f"\nExporting to SLEAP Analysis file...")
	for key, val in data_dict.items():
	print(f"\t{key}: ", end="")
	if isinstance(val, np.ndarray):
	print(f"{val.shape}")
	if transpose:
	# Transpose since MATLAB expects column-major
	f.create_dataset(
	key,
	data=np.transpose(val),
	compression="gzip",
	compression_opts=9,
	)
	else:
	f.create_dataset(
	key, data=val, compression="gzip", compression_opts=9
	)
	else:
	if isinstance(val, (str, int, type(None))):
	print(f"{val}")
	else:
	print(f"{len(val)}")
	f.create_dataset(key, data=val)
	print(f"Saved as {output_path}")

Are you also exporting to analysis in your code somewhere?

There is currently no way of turning off the print statements, but that would be a nice feature. I am moving this over to the Ideas category.

Thanks,
Liezl