diff --git a/hed/models/string_util.py b/hed/models/string_util.py index 589758c3..94acf1da 100644 --- a/hed/models/string_util.py +++ b/hed/models/string_util.py @@ -12,7 +12,7 @@ def gather_descriptions(hed_string): description(str): The concatenated values of all description tags. Side effect: - The input HedString has its Definition tags removed. + The input HedString has its description tags removed. """ desc_tags = hed_string.find_tags("description", recursive=True, include_groups=0) diff --git a/hed/tools/visualization/word_cloud_util.py b/hed/tools/visualization/word_cloud_util.py index 1b022577..d33d06e7 100644 --- a/hed/tools/visualization/word_cloud_util.py +++ b/hed/tools/visualization/word_cloud_util.py @@ -22,7 +22,7 @@ def generate_contour_svg(wc, width, height): contour = _get_contour_mask(wc, width, height) if contour is None: return "" - return _numpy_to_svg(contour) + return _numpy_to_svg(contour, radius=wc.contour_width, color=wc.contour_color) def _get_contour_mask(wc, width, height): @@ -91,19 +91,20 @@ def _draw_contour(wc, img: Image): WordCloud._draw_contour = _draw_contour -def _numpy_to_svg(contour): +def _numpy_to_svg(contour, radius=1, color="black"): """ Convert a numpy array to SVG. Parameters: contour (np.Array): Image to be converted. - + radius (float): The radius of the contour to draw. + color(string): the color to draw it as, e.g. "red" Returns: str: The SVG representation. """ svg_elements = [] points = np.array(contour.nonzero()).T for y, x in points: - svg_elements.append(f'') + svg_elements.append(f'') return '\n'.join(svg_elements)